diff --git a/Text_Based_Chatbot/memorynetwork.py b/Text_Based_Chatbot/memorynetwork.py
index 44ebae75..57d68736 100644
--- a/Text_Based_Chatbot/memorynetwork.py
+++ b/Text_Based_Chatbot/memorynetwork.py
@@ -15,7 +15,7 @@
 from keras.models import Sequential, Model
 from keras.layers.embeddings import Embedding
 from keras.layers import Input, Activation, Dense, Permute, Dropout, merge
-from keras.layers.core import Merge
+from keras.layers import Merge
 from keras.layers import LSTM, GRU
 from keras.utils.data_utils import get_file
 from keras.preprocessing.sequence import pad_sequences
@@ -276,4 +276,4 @@ def vectorize_stories(data, word_idx, story_maxlen, query_maxlen):
         user_prediction = model.predict([user_story, user_query])
         user_prediction = idx_word[np.argmax(user_prediction)]
         print('Result')
-        print(' '.join(user_story_inp), ' '.join(user_query_inp), '| Prediction:', user_prediction)
\ No newline at end of file
+        print(' '.join(user_story_inp), ' '.join(user_query_inp), '| Prediction:', user_prediction)