diff --git a/norch/distributed/distributed.py b/norch/distributed/distributed.py
index b078664..9c797f8 100644
--- a/norch/distributed/distributed.py
+++ b/norch/distributed/distributed.py
@@ -14,11 +14,11 @@ def broadcast_tensor(tensor):
     Tensor._C.broadcast_tensor.argtypes = [ctypes.POINTER(CTensor)]
     Tensor._C.broadcast_tensor.restype = None
     
-    Tensor._C.broadcast_tensor(tensor)
+    Tensor._C.broadcast_tensor(tensor.tensor)
 
 def allreduce_sum_tensor(tensor):
 
     Tensor._C.allreduce_sum_tensor.argtypes = [ctypes.POINTER(CTensor)]
     Tensor._C.allreduce_sum_tensor.restype = None
     
-    Tensor._C.allreduce_sum_tensor(tensor)
+    Tensor._C.allreduce_sum_tensor(tensor.tensor)