diff --git a/ComixGAN/model.py b/ComixGAN/model.py
index d0cba49..96f4473 100644
--- a/ComixGAN/model.py
+++ b/ComixGAN/model.py
@@ -17,6 +17,6 @@ class ComixGAN:
         config.gpu_options.allow_growth = True
         self.session = tf.Session(graph=self.graph, config=config)
         with self.graph.as_default():
-            with tf.device('/GPU:0'):
+            with tf.device('/device:GPU:0'):
                 self.model = load_model(settings.COMIX_GAN_MODEL_PATH,
                                         custom_objects={'InstanceNormalization': InstanceNormalization})