diff --git a/07_Natural_Language_Processing/07_Sentiment_Analysis_With_Doc2Vec/07_sentiment_with_doc2vec.py b/07_Natural_Language_Processing/07_Sentiment_Analysis_With_Doc2Vec/07_sentiment_with_doc2vec.py
index 8d22389fa..ce984f563 100644
--- a/07_Natural_Language_Processing/07_Sentiment_Analysis_With_Doc2Vec/07_sentiment_with_doc2vec.py
+++ b/07_Natural_Language_Processing/07_Sentiment_Analysis_With_Doc2Vec/07_sentiment_with_doc2vec.py
@@ -161,7 +161,7 @@
         for j in range(len(valid_words)):
             valid_word = word_dictionary_rev[valid_examples[j]]
             top_k = 5 # number of nearest neighbors
-            nearest = (-sim[j, :]).argsort()[1:top_k+1]
+            nearest = (-sim[j, :len(word_dictionary_rev)]).argsort()[1:top_k+1]
             log_str = "Nearest to {}:".format(valid_word)
             for k in range(top_k):
                 close_word = word_dictionary_rev[nearest[k]]
@@ -303,4 +303,4 @@
 plt.xlabel('Generation')
 plt.ylabel('Accuracy')
 plt.legend(loc='lower right')
-plt.show()
\ No newline at end of file
+plt.show()