diff --git a/candle-pyo3/test.py b/candle-pyo3/test.py
index aad5e8ae..d63f752b 100644
--- a/candle-pyo3/test.py
+++ b/candle-pyo3/test.py
@@ -5,7 +5,8 @@ print(t)
 print("shape", t.shape, t.rank)
 print(t + t)
 
-t = candle.Tensor([3, 1, 4, 1, 5, 9, 2, 6])
+t = candle.Tensor([3.0, 1, 4, 1, 5, 9, 2, 6])
 print(t)
 print(t+t)
-print(t.reshape([2, 4]))
+t = t.reshape([2, 4])
+print(t.matmul(t.t()))