diff --git a/src/losses.jl b/src/losses.jl
index 40967f1..ca0a7fb 100644
--- a/src/losses.jl
+++ b/src/losses.jl
@@ -50,7 +50,7 @@ import Random
 Random.seed!(123)
 
 const d = 2
-arch = GSympNet(d)
+arch = Chain(Dense(d, d), Dense(d, d))
 nn = NeuralNetwork(arch)
 
 input_vec =  [1., 2.]