diff --git a/test/NN_SDE_tests.jl b/test/NN_SDE_tests.jl
index e410a18f6..608ee33a5 100644
--- a/test/NN_SDE_tests.jl
+++ b/test/NN_SDE_tests.jl
@@ -1,7 +1,6 @@
 @testitem "Test-1" tags=[:nnsde] begin
-    using NeuralPDE, Lux, Optimization, OptimizationOptimJL, Plots
-    using ModelingToolkit: Interval
-    using OrdinaryDiffEq, Random
+    using OrdinaryDiffEq, Random, Lux, Optimisers
+    using OptimizationOptimJL: BFGS
     Random.seed!(100)
 
     α = 1