diff --git a/src/DispersiveShallowWater.jl b/src/DispersiveShallowWater.jl
index f7821acd8..ce6bbb72f 100644
--- a/src/DispersiveShallowWater.jl
+++ b/src/DispersiveShallowWater.jl
@@ -17,7 +17,7 @@ module DispersiveShallowWater
 
 using BandedMatrices: BandedMatrix
 using DelimitedFiles: readdlm
-using DiffEqBase: DiffEqBase, terminate!
+using DiffEqBase: DiffEqBase
 using FastBroadcast: @..
 using ForwardDiff: ForwardDiff
 using Interpolations: Interpolations, linear_interpolation
diff --git a/src/callbacks_step/relaxation.jl b/src/callbacks_step/relaxation.jl
index e73ce42e4..03f611035 100644
--- a/src/callbacks_step/relaxation.jl
+++ b/src/callbacks_step/relaxation.jl
@@ -57,7 +57,7 @@ function (relaxation_callback::RelaxationCallback)(u, t, integrator)
 end
 
 # This method is called as callback during the time integration.
-@inline function (relaxation_callback::RelaxationCallback)(integrator)
+@inline function (relaxation_callback::RelaxationCallback)(integrator::DiffEqBase.DEIntegrator)
     semi = integrator.p
     told = integrator.tprev
     qold = integrator.uprev
@@ -117,7 +117,7 @@ end
         end
 
         if terminate_integration
-            terminate!(integrator)
+            DiffEqBase.terminate!(integrator)
         end
     end
     return nothing