diff --git a/build/distributed.o b/build/distributed.o
index 61ad45c..f9c8535 100644
Binary files a/build/distributed.o and b/build/distributed.o differ
diff --git a/norch/csrc/distributed.cpp b/norch/csrc/distributed.cpp
index 8124ec0..6b25365 100644
--- a/norch/csrc/distributed.cpp
+++ b/norch/csrc/distributed.cpp
@@ -2,7 +2,6 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include "distributed.h"
-#include "tensor.h"
 #include "cuda.h"
 #include <mpi.h>
 #include <nccl.h>
diff --git a/norch/csrc/distributed.h b/norch/csrc/distributed.h
index 1040fc0..1992b8f 100644
--- a/norch/csrc/distributed.h
+++ b/norch/csrc/distributed.h
@@ -1,5 +1,6 @@
 #ifndef DISTRIBUTED_H
 #define DISTRIBUTED_H
+#include "tensor.h"
 
 #define MPI_CHECK(cmd) do {                          \
   int e = cmd;                                      \
diff --git a/norch/libtensor.so b/norch/libtensor.so
index b70187e..9e4f3e6 100755
Binary files a/norch/libtensor.so and b/norch/libtensor.so differ