diff --git a/faiss/gpu/utils/MatrixMult-inl.cuh b/faiss/gpu/utils/MatrixMult-inl.cuh
index 71c66ca623..da4b201f0d 100644
--- a/faiss/gpu/utils/MatrixMult-inl.cuh
+++ b/faiss/gpu/utils/MatrixMult-inl.cuh
@@ -168,8 +168,8 @@ void runMatrixMult(
     FAISS_ASSERT(c.getSize(0) <= std::numeric_limits<int>::max());
     FAISS_ASSERT(c.getSize(1) <= std::numeric_limits<int>::max());
 
-    FAISS_ASSERT(a.getSize(0) <= std::numeric_limits<int>::max());
-    FAISS_ASSERT(a.getSize(1) <= std::numeric_limits<int>::max());
+    FAISS_ASSERT(b.getSize(0) <= std::numeric_limits<int>::max());
+    FAISS_ASSERT(b.getSize(1) <= std::numeric_limits<int>::max());
 
     FAISS_ASSERT(a.getSize(0) <= std::numeric_limits<int>::max());
     FAISS_ASSERT(a.getSize(1) <= std::numeric_limits<int>::max());