diff --git a/terra-jupyter-python/tests/smoke_test.ipynb b/terra-jupyter-python/tests/smoke_test.ipynb
index a3f33b3d..862eeb55 100644
--- a/terra-jupyter-python/tests/smoke_test.ipynb
+++ b/terra-jupyter-python/tests/smoke_test.ipynb
@@ -248,14 +248,14 @@
    "source": [
     "import numpy as np\n",
     "import pandas as pd\n",
-    "import ydata_profiling\n",
+    "from ydata_profiling import ProfileReport\n",
     "\n",
     "df = pd.DataFrame(\n",
     "    np.random.rand(100, 5),\n",
     "    columns=['a', 'b', 'c', 'd', 'e']\n",
     ")\n",
     "\n",
-    "profile = ydata_profiling(df, title='Pandas Profiling Report')\n",
+    "profile = ProfileReport(df, title='Pandas Profiling Report')\n",
     "profile"
    ]
   },