diff --git a/notebooks/from-zero-to-hero-tutorial/02_models.ipynb b/notebooks/from-zero-to-hero-tutorial/02_models.ipynb
index c7667f334..381d2e7ef 100644
--- a/notebooks/from-zero-to-hero-tutorial/02_models.ipynb
+++ b/notebooks/from-zero-to-hero-tutorial/02_models.ipynb
@@ -34,7 +34,7 @@
    },
    "outputs": [],
    "source": [
-    "!pip install avalanche-lib==0.5"
+    "!pip install avalanche-lib==0.6"
    ]
   },
   {