diff --git a/Phase 2/task_7.ipynb b/Phase 2/task_7.ipynb
index bc31b2f..8ff27dd 100644
--- a/Phase 2/task_7.ipynb	
+++ b/Phase 2/task_7.ipynb	
@@ -163,7 +163,7 @@
     "        image_fd = np.array(get_all_fd(image_id)[selected_feature_model]).flatten()\n",
     "    else:\n",
     "        image_fd = np.array(\n",
-    "            fd_collection.find({\"image_id\": image_id})[selected_feature_model]\n",
+    "            fd_collection.find_one({\"image_id\": image_id})[selected_feature_model]\n",
     "        ).flatten()\n",
     "\n",
     "    match dim_reduction:\n",
diff --git a/Phase 2/task_8.ipynb b/Phase 2/task_8.ipynb
index fd289ae..a3fac6d 100644
--- a/Phase 2/task_8.ipynb	
+++ b/Phase 2/task_8.ipynb	
@@ -163,7 +163,7 @@
     "        image_fd = np.array(get_all_fd(image_id)[selected_feature_model]).flatten()\n",
     "    else:\n",
     "        image_fd = np.array(\n",
-    "            fd_collection.find({\"image_id\": image_id})[selected_feature_model]\n",
+    "            fd_collection.find_one({\"image_id\": image_id})[selected_feature_model]\n",
     "        ).flatten()\n",
     "\n",
     "    match dim_reduction:\n",