diff --git a/tests/conftest.py b/tests/conftest.py
index d06cc0f71..1fdf7c9f1 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -771,7 +771,7 @@ def ar_vid_path(tmp_session) -> str:
 def ar_milk_bottle_path(tmp_session) -> str:
     """ Returns the path of the milk bottle action dataset. """
     return unzip_url(
-        ar_urls.milk_bottle_action_path,
+        ar_urls.milk_bottle_action_minified_path,
         fpath=tmp_session,
         dest=tmp_session,
         exist_ok=True,
diff --git a/tests/unit/action_recognition/test_action_recognition_notebooks.py b/tests/unit/action_recognition/test_action_recognition_notebooks.py
index 3e4e3f983..134195efc 100644
--- a/tests/unit/action_recognition/test_action_recognition_notebooks.py
+++ b/tests/unit/action_recognition/test_action_recognition_notebooks.py
@@ -8,6 +8,7 @@
 import pytest
 import scrapbook as sb
 
+from utils_cv.common.data import unzip_url
 from utils_cv.action_recognition.data import Urls
 
 # Unless manually modified, python3 should be
@@ -33,15 +34,19 @@ def test_00_notebook_run(action_recognition_notebooks):
 
 
 @pytest.mark.notebooks
-def test_01_notebook_run(action_recognition_notebooks):
+def test_01_notebook_run(
+    action_recognition_notebooks,
+    ar_milk_bottle_path
+):
     notebook_path = action_recognition_notebooks["01"]
     pm.execute_notebook(
         notebook_path,
         OUTPUT_NOTEBOOK,
         parameters=dict(
             PM_VERSION=pm.__version__,
+            DATA_PATH=ar_milk_bottle_path,
             MODEL_INPUT_SIZE=8,
-            EPOCHS=2,
+            EPOCHS=1,
             BATCH_SIZE=8,
             LR=0.001,
         ),
diff --git a/utils_cv/action_recognition/data.py b/utils_cv/action_recognition/data.py
index 524a469fb..a4681ca95 100644
--- a/utils_cv/action_recognition/data.py
+++ b/utils_cv/action_recognition/data.py
@@ -54,6 +54,9 @@ class Urls:
 
     # testing datasets
     milk_bottle_action_path = urljoin(base, "milkBottleActions.zip")
+    milk_bottle_action_minified_path = urljoin(
+        base, "milkBottleActions_minified.zip"
+    )
 
     # milk bottle action split test files
     milk_bottle_action_train_split = urljoin(