diff --git a/keras.py b/keras.py
index d1cc70824aebe72ee15de5c8c2aeb32176b035e9..8668f2e56344d9feda67b273408298d2ebaa98ed 100644
--- a/keras.py
+++ b/keras.py
@@ -1,4 +1,4 @@
-from itertools import izip
+#from itertools import izip
 from collections import OrderedDict, defaultdict
 import fnmatch
 import numpy as np
@@ -7,12 +7,12 @@ from tqdm import tqdm
 from inspect import getargspec
 from warnings import warn
 import re
-from tensorflow.python.keras.engine.training_utils import make_logs
+#from tensorflow.python.keras.engine.training_utils import make_logs
 from tensorflow.python.keras.backend import track_variable
 from operator import itemgetter
 
-from evil import pin
-from data import SKDict, DSS
+from .evil import pin
+from .data import SKDict, DSS
 
 # various helper functions
 
@@ -188,7 +188,7 @@ class Moment2Std(tf.keras.callbacks.Callback):
     def on_x_end(self, x, logs=None):
         if logs is None:
             return
-        for key1, mom1 in logs.items():
+        for key1, mom1 in list(logs.items()):
             if not key1.endswith(self.mom1):
                 continue
             prefix = key1[:-len(self.mom1)]