1 files changed, 6 insertions, 8 deletions
diff --git a/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp b/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp
index 6a47e0da5df..56b9473b6e6 100644
--- a/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp
+++ b/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp
@@ -450,22 +450,20 @@ DenseTensorAttribute::onCommit()
 }
 
 void
-DenseTensorAttribute::onGenerationChange(generation_t next_gen)
+DenseTensorAttribute::before_inc_generation(generation_t current_gen)
 {
-    // TODO: Change onGenerationChange() to send current generation instead of next generation.
-    //       This applies for entire attribute vector code.
-    TensorAttribute::onGenerationChange(next_gen);
+    TensorAttribute::before_inc_generation(current_gen);
     if (_index) {
-        _index->assign_generation(next_gen - 1);
+        _index->assign_generation(current_gen);
     }
 }
 
 void
-DenseTensorAttribute::removeOldGenerations(generation_t first_used_gen)
+DenseTensorAttribute::reclaim_memory(generation_t oldest_used_gen)
 {
-    TensorAttribute::removeOldGenerations(first_used_gen);
+    TensorAttribute::reclaim_memory(oldest_used_gen);
     if (_index) {
-        _index->reclaim_memory(first_used_gen);
+        _index->reclaim_memory(oldest_used_gen);
     }
 }