diff --git a/sc2bench/__init__.py b/sc2bench/__init__.py
index 32f17c3d..e7681732 100644
--- a/sc2bench/__init__.py
+++ b/sc2bench/__init__.py
@@ -1,3 +1,3 @@
 from . import models, transforms, loss
 
-__version__ = '0.0.3'
+__version__ = '0.0.4'
diff --git a/setup.py b/setup.py
index 048e35c1..f8d0d4c9 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 description = 'SC2 Benchmark: Supervised Compression for Split Computing.'
 setup(
     name='sc2bench',
-    version='0.0.3',
+    version='0.0.4',
     author='Yoshitomo Matsubara',
     description=description,
     long_description=long_description,
@@ -22,7 +22,7 @@
         'scipy',
         'cython',
         'pycocotools>=2.0.2',
-        'torchdistill>=0.2.7',
+        'torchdistill<=0.3.3',
         'compressai>=1.1.8,<=1.2.2',
         'timm>=0.4.12'
     ],