diff --git a/fasttext_amazon_reviews/fasttext_amazon_reviews.ipynb b/fasttext_amazon_reviews/fasttext_amazon_reviews.ipynb
index 6dbe47b..a89a973 100644
--- a/fasttext_amazon_reviews/fasttext_amazon_reviews.ipynb
+++ b/fasttext_amazon_reviews/fasttext_amazon_reviews.ipynb
@@ -48,7 +48,7 @@
    "outputs": [],
    "source": [
     "import json\n",
-    "from cleanlab.models.fasttext import FastTextClassifier, data_loader\n",
+    "from utils.fasttext_wrapper import FastTextClassifier, data_loader\n",
     "import cleanlab\n",
     "import numpy as np\n",
     "from sklearn.model_selection import StratifiedKFold\n",