diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index b516b4e..db41203 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install tensorflow==2.11.0 efficientnet==1.0.8
+          pip install tensorflow==2.11.0
           pip install -e .
 
       - name: Run tests
diff --git a/tests/test_predict.py b/tests/test_predict.py
index f244b03..ef8cb72 100644
--- a/tests/test_predict.py
+++ b/tests/test_predict.py
@@ -3,7 +3,7 @@
 import tempfile
 import unittest
 
-import efficientnet.keras as efn
+# import efficientnet.keras as efn
 import requests
 
 from predictor import predict