diff --git a/Project.toml b/Project.toml
index 81acac6c1..0e02eea04 100644
--- a/Project.toml
+++ b/Project.toml
@@ -61,7 +61,7 @@ OrdinaryDiffEqCore = "1"
 OrdinaryDiffEqTsit5 = "1"
 Pkg = "1"
 Random = "1"
-SciMLBase = "2"
+SciMLBase = "2.105"
 SciMLOperators = "1.4"
 SparseArrays = "1"
 SpecialFunctions = "2"
diff --git a/src/time_evolution/time_evolution.jl b/src/time_evolution/time_evolution.jl
index d04c7338b..3b09bd443 100644
--- a/src/time_evolution/time_evolution.jl
+++ b/src/time_evolution/time_evolution.jl
@@ -504,14 +504,6 @@ end
     end
 end
 
-#TODO: Remove when a new release of SciMLBase.jl >2.104.0 is available
-(f::SDEFunction)(du, u, p, t) =
-    if f.f isa AbstractSciMLOperator
-        f.f(du, u, u, p, t)
-    else
-        f.f(du, u, p, t)
-    end
-
 #######################################
 
 function liouvillian_floquet(