diff --git a/src/index.cpp b/src/index.cpp
index 67254fff1..e6be778c2 100644
--- a/src/index.cpp
+++ b/src/index.cpp
@@ -659,7 +659,7 @@ void Index<T, TagT, LabelT>::load(const char *filename, uint32_t num_threads, ui
     
     reposition_frozen_point_to_end();
 
-    _table_stats.tag_memory_usage = _table_stats.node_mem_usage
+    _table_stats.total_mem_usage = _table_stats.node_mem_usage
         + _table_stats.graph_mem_usage
         + _table_stats.label_mem_usage
         + _table_stats.tag_memory_usage;
diff --git a/src/pq_flash_index.cpp b/src/pq_flash_index.cpp
index 69386c7db..bfcc65487 100644
--- a/src/pq_flash_index.cpp
+++ b/src/pq_flash_index.cpp
@@ -1232,7 +1232,7 @@ int PQFlashIndex<T, LabelT>::load_from_separate_paths(uint32_t num_threads, cons
         delete[] norm_val;
     }
 
-    _table_stats.tag_memory_usage = _table_stats.node_mem_usage
+    _table_stats.total_mem_usage = _table_stats.node_mem_usage
         + _table_stats.graph_mem_usage
         + _table_stats.label_mem_usage
         + _table_stats.tag_memory_usage;