diff --git a/examples/cmaes.ipynb b/examples/cmaes.ipynb
index ba059cdc..023b2e27 100644
--- a/examples/cmaes.ipynb
+++ b/examples/cmaes.ipynb
@@ -55,7 +55,7 @@
     "import matplotlib.pyplot as plt\n",
     "from matplotlib.patches import Ellipse\n",
     "\n",
-    "from qdax.core.cmaes import CMAES"
+    "from qdax.baselines.cmaes import CMAES"
    ]
   },
   {