diff --git a/src/ivfkmeans.c b/src/ivfkmeans.c
index 83e534f..dc09552 100644
--- a/src/ivfkmeans.c
+++ b/src/ivfkmeans.c
@@ -280,7 +280,7 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers, const Ivff
 	IvfflatCheckMemoryUsage(totalSize);
 
 	/* Ensure indexing does not overflow */
-	if (numCenters * numCenters > INT_MAX)
+	if (numCenters > INT_MAX / numCenters)
 		elog(ERROR, "Indexing overflow detected. Please report a bug.");
 
 	/* Set support functions */