diff --git a/neuralintents/main.py b/neuralintents/main.py
index 9d24bc2..d5870cc 100644
--- a/neuralintents/main.py
+++ b/neuralintents/main.py
@@ -13,7 +13,7 @@
 
 from tensorflow.keras.models import Sequential
 from tensorflow.keras.layers import Dense, Dropout
-from tensorflow.keras.optimizers import SGD
+from tensorflow.keras.optimizers.legacy import SGD
 from tensorflow.keras.models import load_model
 
 nltk.download('punkt', quiet=True)
@@ -184,4 +184,4 @@ def request(self, message):
         if ints[0]['intent'] in self.intent_methods.keys():
             self.intent_methods[ints[0]['intent']]()
         else:
-            return self._get_response(ints, self.intents)
\ No newline at end of file
+            return self._get_response(ints, self.intents)