diff --git a/n2v/models/n2v_standard.py b/n2v/models/n2v_standard.py
index 4bd6ecc..dc4d2f9 100644
--- a/n2v/models/n2v_standard.py
+++ b/n2v/models/n2v_standard.py
@@ -304,7 +304,7 @@ def prepare_for_training(self, optimizer=None, **kwargs):
                     TensorBoard(log_dir=str(self.logdir / 'logs'), write_graph=False, profile_batch=0))
 
         if self.config.train_reduce_lr is not None:
-            from keras.callbacks import ReduceLROnPlateau
+            from tensorflow.keras.callbacks import ReduceLROnPlateau
             rlrop_params = self.config.train_reduce_lr
             if 'verbose' not in rlrop_params:
                 rlrop_params['verbose'] = True
diff --git a/n2v/version.py b/n2v/version.py
index 0404d81..e1424ed 100644
--- a/n2v/version.py
+++ b/n2v/version.py
@@ -1 +1 @@
-__version__ = '0.3.0'
+__version__ = '0.3.1'