diff --git a/gan/gan.py b/gan/gan.py
index b7033d44f6..c850cdd259 100644
--- a/gan/gan.py
+++ b/gan/gan.py
@@ -6,7 +6,7 @@
 from keras.layers.advanced_activations import LeakyReLU
 from keras.layers.convolutional import UpSampling2D, Conv2D
 from keras.models import Sequential, Model
-from keras.optimizers import Adam
+from tensorflow.keras.optimizers import Adam
 
 import matplotlib.pyplot as plt