diff --git a/Project.toml b/Project.toml
index b6fd056..dfebd97 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJIteration"
 uuid = "614be32b-d00c-4edb-bd02-1eb411ab5e55"
 authors = ["Anthony D. Blaom <anthony.blaom@gmail.com>"]
-version = "0.4.0"
+version = "0.4.1"
 
 [deps]
 IterationControl = "b3c1a2ee-3fec-4384-bf48-272ea71de57c"
@@ -10,7 +10,7 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 
 [compat]
 IterationControl = "0.5"
-MLJBase = "0.18.8"
+MLJBase = "0.18.8, 0.19"
 julia = "1"
 
 [extras]
diff --git a/test/controls.jl b/test/controls.jl
index ca452ac..1beb432 100644
--- a/test/controls.jl
+++ b/test/controls.jl
@@ -143,7 +143,7 @@ const N = 20
                 local model = DummyIterativeModel(n=n)
                 local mach = machine(model, X, y)
                 fit!(mach, verbosity=0)
-                t = mach |> EXT_GIVEN_STR[$str] |> PROJECTION_GIVEN_STR[$str]
+                t = PROJECTION_GIVEN_STR[$str](mach |> EXT_GIVEN_STR[$str])
                 push!(trace_by_hand, t)
             end