diff --git a/setup.py b/setup.py
index fb06545..587ff84 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
     "scikit-learn",
     "torch",
     "torchvision",
-    "tinyfk>=0.4.8",
+    "tinyfk<0.6",
     "tqdm",
     "matplotlib",
     "albumentations",