diff --git a/test/integrators/cache.jl b/test/integrators/cache.jl
index ff42b74..bdd5359 100644
--- a/test/integrators/cache.jl
+++ b/test/integrators/cache.jl
@@ -41,14 +41,14 @@ end
 println("Check some other integrators")
 
 println("Rosenbrock23")
-@test_nowarn solve(prob, MethodOfSteps(Rosenbrock23(chunk_size = 1)); dt = 0.5)
+@test_nowarn solve(prob, MethodOfSteps(Rosenbrock23()); dt = 0.5)
 
 @testset "$(nameof(typeof(alg)))" for alg in CACHE_TEST_ALGS
     @test_nowarn solve(prob, MethodOfSteps(alg); dt = 0.5)
 end
 
 println("Rodas4")
-@test_nowarn solve(prob, MethodOfSteps(Rodas4(chunk_size = 1)); dt = 0.5)
+@test_nowarn solve(prob, MethodOfSteps(Rodas4()); dt = 0.5)
 
 println("Rodas5")
-@test_nowarn solve(prob, MethodOfSteps(Rodas5(chunk_size = 1)); dt = 0.5)
+@test_nowarn solve(prob, MethodOfSteps(Rodas5()); dt = 0.5)
diff --git a/test/integrators/initialization.jl b/test/integrators/initialization.jl
index 3d94453..d1fd1df 100644
--- a/test/integrators/initialization.jl
+++ b/test/integrators/initialization.jl
@@ -1,5 +1,4 @@
 using DelayDiffEq
-using SciMLBase
 using LinearAlgebra
 using Test
 
@@ -26,13 +25,13 @@ using Test
     prob_ddae = DDEProblem(
         DDEFunction{true}(sir_ddae!;
             mass_matrix = Diagonal([1.0, 1.0, 0.0])),
-        u0,
+        u0_good,
         sir_history,
         tspan,
         p;
         constant_lags = (p.τ,))
     alg = MethodOfSteps(Rosenbrock23())
     @test_nowarn init(prob_ddae, alg)
-    prob.u0[1] = 2.0
+    prob_ddae.u0[1] = 2.0
     @test_throws SciMLBase.CheckInitFailureError init(prob_ddae, alg)
 end