diff --git a/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp b/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
index 5ece23c7aad0aeb06860734a287509c231523fbf..5f787a415c45bc01c57496981f97e376d2916c5b 100644
--- a/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
+++ b/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
@@ -540,7 +540,7 @@ TEST(FAISSTEST, sift1m_L2) {
     test_ann_hdf5("glove-200-angular", "IVF16384,SQ8",       1, {8, 128});
     test_ann_hdf5("glove-200-angular", "IVF16384,SQ8Hybrid", 1, {8, 128});
 #ifdef CUSTOMIZATION
-    test_ivfsq8h_gpu("glove-200-angular", 2, {128, 1024});
+    test_ivfsq8h_gpu("glove-200-angular", 2, {8, 128});
 #endif
 }