diff --git a/Project.toml b/Project.toml
index f60277dab..2520589b5 100644
--- a/Project.toml
+++ b/Project.toml
@@ -87,7 +87,7 @@ MLDataDevices = "1.2.0"
 MethodOfLines = "0.11.6"
 ModelingToolkit = "9.46, 10"
 MonteCarloMeasurements = "1.1"
-NeuralOperators = "0.5"
+NeuralOperators = "0.5, 0.6"
 Optimisers = "0.3.3, 0.4"
 Optimization = "4"
 OptimizationOptimJL = "0.4"