diff --git a/pygrad/tensor.py b/pygrad/tensor.py
index 54a7e6b..b424390 100644
--- a/pygrad/tensor.py
+++ b/pygrad/tensor.py
@@ -27,7 +27,7 @@ class Tensor:
     tensor._save(self, other)
 
     def back(upstream):
-      return np.dot(np.ones_like(self.value), upstream), np.dot(np.ones_like(self.value), upstream)
+      return np.dot(np.ones_like(self.value).T, upstream), np.dot(np.ones_like(self.value).T, upstream)
 
     tensor._back = back
     return tensor