diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index c1b12b0..e58880f 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -24,7 +24,7 @@ jobs:
           cache: 'pip'
       - run: pip install -U flake8
       - name: Run python code lint
-        run: flake8 moltx
+        run: flake8 dilipredict
 
   tests:
     name: "Python ${{ matrix.py-version }} unit-tests"
diff --git a/requirements.txt b/requirements.txt
index 91ac4e5..faadcff 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,6 @@
-einops
+numpy==1.24.4
 torch==1.13.1
 torchvision==0.14.1
+einops
+pillow
+scikit-learn
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
deleted file mode 100644
index e69de29..0000000
diff --git a/tests/test_piplines.py b/tests/test_piplines.py
index 2b97957..617e076 100644
--- a/tests/test_piplines.py
+++ b/tests/test_piplines.py
@@ -1,7 +1,7 @@
 from dilipredict import models, pipelines
 
 
-def test_MutSmiReg(image_loader):
+def test_DILIPredict(image_loader):
     test_img_path = 'tests/data/test_img.tiff'
     image_files = [
         [test_img_path] * 10,