diff --git a/exercises/04-create-embeddings.ipynb b/exercises/04-create-embeddings.ipynb
index 7d1d2f0..8bd3787 100644
--- a/exercises/04-create-embeddings.ipynb
+++ b/exercises/04-create-embeddings.ipynb
@@ -94,7 +94,7 @@
     "\n",
     "# TODO the get_cosine_similarity function does not work very well does it? Fix it!\n",
     "def get_cosine_similarity(vector_1, vector_2):\n",
-    "    return 1 - spatial.distance.cosine(vector_1, vector_1)"
+    "    return 1 - spatial.distance.cosine(vector_1, vector_2)"
    ]
   },
   {