diff --git a/run.py b/run.py
index 31bc6da..95d679c 100644
--- a/run.py
+++ b/run.py
@@ -2,5 +2,8 @@
 
 from modules import core
 
+import os
+os.environ['KERAS_BACKEND'] = 'tensorflow'
+
 if __name__ == '__main__':
     core.run()