diff --git a/.github/codecov.yml b/.github/codecov.yml
new file mode 100644
index 0000000..ed9d9f1
--- /dev/null
+++ b/.github/codecov.yml
@@ -0,0 +1,8 @@
+coverage:
+  status:
+    project:
+      default:
+        threshold: 0.5%
+    patch:
+      default:
+        target: 80%
\ No newline at end of file
diff --git a/Project.toml b/Project.toml
index 8366021..d1e69f8 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "IterationControl"
 uuid = "b3c1a2ee-3fec-4384-bf48-272ea71de57c"
 authors = ["Anthony D. Blaom <anthony.blaom@gmail.com>"]
-version = "0.5.3"
+version = "0.5.4"
 
 [deps]
 EarlyStopping = "792122b4-ca99-40de-a6bc-6742525f08b6"
diff --git a/src/stopping_controls.jl b/src/stopping_controls.jl
index 867fe6a..f6cab9d 100644
--- a/src/stopping_controls.jl
+++ b/src/stopping_controls.jl
@@ -29,7 +29,7 @@ for ex in EARLY_STOPPING_WRAPPERS
         needs_loss(wrapper::$ex) =
             needs_loss(wrapper.criterion)
         needs_training_losses(wrapper::$ex) =
-            needs_training(wrapper.criterion)
+            needs_training_losses(wrapper.criterion)
     end |> eval
 end