diff --git a/docs/src/neural.md b/docs/src/neural.md
index 6b3c98816..551c8994e 100644
--- a/docs/src/neural.md
+++ b/docs/src/neural.md
@@ -52,15 +52,15 @@ As always, getting the model right is hardest thing.
 
 ```@example Neural_surrogate
 model1 = Chain(
-  Dense(1, 5, σ),
+  Dense(2, 5, σ),
   Dense(5,2,σ),
   Dense(2, 1)
 )
-neural = NeuralSurrogate(x, y, lb, ub, model = model1, n_echos = 10)
+neural = NeuralSurrogate(xys, zs, lower_bound, upper_bound, model = model1, n_echos = 10)
 ```
 
 ## Optimization
 We can now call an optimization function on the neural network:
 ```@example Neural_surrogate
-surrogate_optimize(schaffer, SRBF(), lower_bound, upper_bound, neura, SobolSample(), maxiters=20, num_new_samples=10)
+surrogate_optimize(schaffer, SRBF(), lower_bound, upper_bound, neural, SobolSample(), maxiters=20, num_new_samples=10)
 ```