diff --git a/kraken/ketos/pretrain.py b/kraken/ketos/pretrain.py
index db21546e3..f323cd9f0 100644
--- a/kraken/ketos/pretrain.py
+++ b/kraken/ketos/pretrain.py
@@ -290,7 +290,7 @@ def pretrain(ctx, batch_size, pad, output, spec, load, freq, quit, epochs,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)
 
diff --git a/kraken/ketos/recognition.py b/kraken/ketos/recognition.py
index 0c8d4188d..63dcc8f70 100644
--- a/kraken/ketos/recognition.py
+++ b/kraken/ketos/recognition.py
@@ -324,7 +324,7 @@ def train(ctx, batch_size, pad, output, spec, append, load, freq, quit, epochs,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)
 
diff --git a/kraken/ketos/ro.py b/kraken/ketos/ro.py
index 65f93a8be..4075b638a 100644
--- a/kraken/ketos/ro.py
+++ b/kraken/ketos/ro.py
@@ -260,7 +260,7 @@ def rotrain(ctx, batch_size, output, load, freq, quit, epochs, min_epochs, lag,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)
 
diff --git a/kraken/ketos/segmentation.py b/kraken/ketos/segmentation.py
index 24a91735d..711aab1da 100644
--- a/kraken/ketos/segmentation.py
+++ b/kraken/ketos/segmentation.py
@@ -357,7 +357,7 @@ def segtrain(ctx, output, spec, line_width, pad, load, freq, quit, epochs,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)