1 files changed, 5 insertions, 5 deletions
diff --git a/storage/src/tests/distributor/top_level_distributor_test_util.h b/storage/src/tests/distributor/top_level_distributor_test_util.h
index e8794eb4199..6efc36a8215 100644
--- a/storage/src/tests/distributor/top_level_distributor_test_util.h
+++ b/storage/src/tests/distributor/top_level_distributor_test_util.h
@@ -17,7 +17,7 @@ namespace framework { struct TickingThreadPool; }
 
 namespace distributor {
 
-class TopLevelDistributor;
+class BucketSpaceStateMap;
 class DistributorBucketSpace;
 class DistributorBucketSpaceRepo;
 class DistributorMetricSet;
@@ -26,10 +26,11 @@ class DistributorStripe;
 class DistributorStripeComponent;
 class DistributorStripeOperationContext;
 class DistributorStripePool;
-class StripeAccessGuard;
 class IdealStateMetricSet;
 class Operation;
+class StripeAccessGuard;
 class TopLevelBucketDBUpdater;
+class TopLevelDistributor;
 
 class TopLevelDistributorTestUtil : private DoneInitializeHandler
 {
@@ -60,9 +61,8 @@ public:
     // As the above, but always inserts into default bucket space
     void add_nodes_to_stripe_bucket_db(const document::BucketId& id, const std::string& nodeStr);
 
-    // TODO STRIPE replace with BucketSpaceStateMap once legacy is gone
-    DistributorBucketSpaceRepo& top_level_bucket_space_repo() noexcept;
-    const DistributorBucketSpaceRepo& top_level_bucket_space_repo() const noexcept;
+    BucketSpaceStateMap& bucket_space_states() noexcept;
+    const BucketSpaceStateMap& bucket_space_states() const noexcept;
 
     std::unique_ptr<StripeAccessGuard> acquire_stripe_guard();