diff --git a/test/TensorFlowNET.Examples/BasicModels/NeuralNetXor.cs b/test/TensorFlowNET.Examples/BasicModels/NeuralNetXor.cs
index e9bcf0cb..fb40157d 100644
--- a/test/TensorFlowNET.Examples/BasicModels/NeuralNetXor.cs
+++ b/test/TensorFlowNET.Examples/BasicModels/NeuralNetXor.cs
@@ -54,7 +54,7 @@ namespace TensorFlowNET.Examples
             var logits = tf.matmul(hidden_activations, output_weights);
 
             // Shape [4]
-            var predictions = tf.sigmoid(tf.squeeze(logits));
+            var predictions = tf.tanh(tf.squeeze(logits));
             var loss = tf.reduce_mean(tf.square(predictions - tf.cast(labels, tf.float32)), name:"loss");
 
             var gs = tf.Variable(0, trainable: false, name: "global_step");