diff --git a/tests/allgather_test.cu b/tests/allgather_test.cu
index d26e30f6..1bb0635a 100644
--- a/tests/allgather_test.cu
+++ b/tests/allgather_test.cu
@@ -180,7 +180,7 @@ int main(int argc, const char *argv[])
 
   int *data_d;
   uint64_t *flag_d;
-  size_t data_size = 1024*1024*1024;
+  size_t data_size = 1024*1024;
   int nelemsPerGPU = data_size / sizeof(int) / world_size;
   CUDACHECK(cudaMalloc(&data_d, data_size));
   CUDACHECK(cudaMalloc(&flag_d, sizeof(uint64_t)));