diff --git a/src/runtime/graph/graph_runtime.cc b/src/runtime/graph/graph_runtime.cc
index ed833d40848c1b3bf92de1a1ea6fe3f8c99fe97b..d244fe5f028e7ed065e45f2796a130cfb4e18ac5 100644
--- a/src/runtime/graph/graph_runtime.cc
+++ b/src/runtime/graph/graph_runtime.cc
@@ -386,7 +386,7 @@ void GraphRuntime::LoadDLTensor(dmlc::Stream* strm, DLTensor* dst) {
   for (int i = 0; i < dst->ndim; ++i) {
     size *= dst->shape[i];
   }
-  int64_t data_byte_size;
+  uint64_t data_byte_size;
   CHECK(strm->Read(&data_byte_size, sizeof(data_byte_size)))
       << "Invalid DLTensor file format";
   CHECK(data_byte_size == size)