diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 4c0f64b..bc682e1 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -42,7 +42,7 @@ jobs:
       with:
         name: model
         path: |
-          ./github_project/hp_model.h5
+          ./github_project
 
   evaluate:
     needs: train
@@ -66,7 +66,7 @@ jobs:
       uses: actions/download-artifact@v2
       with:
         name: model
-        path: ./github_project/hp_model.h5
+        path: ./github_project
 
     - name: Evaluate Model
       run: python ./github_project/evaluate.py ${{ github.run_number }}