diff --git a/Sklearn Basics 2 - Train a Classifier on a Star Multi-Table Dataset.ipynb b/Sklearn Basics 2 - Train a Classifier on a Star Multi-Table Dataset.ipynb
index f6ca53d..1dd6084 100644
--- a/Sklearn Basics 2 - Train a Classifier on a Star Multi-Table Dataset.ipynb	
+++ b/Sklearn Basics 2 - Train a Classifier on a Star Multi-Table Dataset.ipynb	
@@ -21,8 +21,7 @@
     "import os\n",
     "import pandas as pd\n",
     "from khiops import core as kh\n",
-    "from khiops.sklearn import KhiopsClassifier\n",
-    "from khiops.utils.helpers import train_test_split_dataset\n",
+    "from khiops.sklearn import KhiopsClassifier, train_test_split_dataset\n",
     "from sklearn import metrics\n",
     "\n",
     "# If there are any issues you may Khiops status with the following command\n",
@@ -529,4 +528,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
\ No newline at end of file
+}
diff --git a/Sklearn Basics 3 - Train a Classifier on a Snowflake Multi-Table Dataset.ipynb b/Sklearn Basics 3 - Train a Classifier on a Snowflake Multi-Table Dataset.ipynb
index 8a1f4b2..b76f232 100644
--- a/Sklearn Basics 3 - Train a Classifier on a Snowflake Multi-Table Dataset.ipynb	
+++ b/Sklearn Basics 3 - Train a Classifier on a Snowflake Multi-Table Dataset.ipynb	
@@ -20,8 +20,7 @@
     "import os\n",
     "import pandas as pd\n",
     "from khiops import core as kh\n",
-    "from khiops.sklearn import KhiopsClassifier\n",
-    "from khiops.utils.helpers import train_test_split_dataset\n",
+    "from khiops.sklearn import KhiopsClassifier, train_test_split_dataset\n",
     "from sklearn import metrics\n",
     "\n",
     "# If there are any issues you may Khiops status with the following command\n",