diff --git a/src/ivfkmeans.c b/src/ivfkmeans.c
index ab8f7d4..003d55d 100644
--- a/src/ivfkmeans.c
+++ b/src/ivfkmeans.c
@@ -327,7 +327,7 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers, const Ivff
 	newCenters->length = numCenters;
 
 #ifdef IVFFLAT_MEMORY
-	ShowMemoryUsage(MemoryContextGetParent(CurrentMemoryContext));
+	ShowMemoryUsage(MemoryContextGetParent(CurrentMemoryContext), totalSize);
 #endif
 
 	/* Pick initial centers */