diff --git a/pygrad/nn.py b/pygrad/nn.py
index d17a42a..a06366c 100644
--- a/pygrad/nn.py
+++ b/pygrad/nn.py
@@ -48,7 +48,7 @@ def single_pass():
 # Small test to see if autograd works.
 def test():
   # Input tensors.
-  x, y, z = tensor.Tensor(-2), tensor.Tensor(5), tensor.Tensor(-4)
+  x, y, z = tensor.Tensor(np.array([[1, 2, 3]])), tensor.Tensor(np.array([[2, 3, 4]])), tensor.Tensor(np.array([[1], [2], [3]]))
   
   # Forward pass.
   q = x.add(y)