diff --git a/examples/paper/continuum_neural_network.py b/examples/paper/continuum_neural_network.py
index 11418063..87449403 100644
--- a/examples/paper/continuum_neural_network.py
+++ b/examples/paper/continuum_neural_network.py
@@ -238,7 +238,7 @@ def load(self, filename):
     # clf = PLSRegression(max_degree + 1)
     # clf.fit(XS_train, y_train)
     clf = ContinuumModel(verbose=True, loss="mse")
-    history = clf.fit(XS_train, y_train, validation_data=(X_validate, y_validate))
+    history = clf.fit(X_train, y_train, validation_data=(X_validate, y_validate))
 
     clf.model.save("model.dat")