1 files changed, 3 insertions, 6 deletions
diff --git a/storage/src/vespa/storage/distributor/distributor_operation_context.h b/storage/src/vespa/storage/distributor/distributor_operation_context.h
index e0d481a322a..9dd853c7e46 100644
--- a/storage/src/vespa/storage/distributor/distributor_operation_context.h
+++ b/storage/src/vespa/storage/distributor/distributor_operation_context.h
@@ -6,10 +6,10 @@
 #include <vespa/storageapi/defs.h>
 
 namespace storage { class DistributorConfiguration; }
-namespace storage::lib { class ClusterStateBundle; }
 
 namespace storage::distributor {
 
+class BucketSpaceStateMap;
 class DistributorBucketSpaceRepo;
 
 /**
@@ -19,11 +19,8 @@ class DistributorOperationContext {
 public:
     virtual ~DistributorOperationContext() {}
     virtual api::Timestamp generate_unique_timestamp() = 0;
-    // TODO STRIPE: Access to bucket space repos is only temporary at this level.
-    virtual const DistributorBucketSpaceRepo& bucket_space_repo() const noexcept = 0;
-    virtual DistributorBucketSpaceRepo& bucket_space_repo() noexcept = 0;
-    virtual const DistributorBucketSpaceRepo& read_only_bucket_space_repo() const noexcept = 0;
-    virtual DistributorBucketSpaceRepo& read_only_bucket_space_repo() noexcept = 0;
+    virtual const BucketSpaceStateMap& bucket_space_states() const noexcept = 0;
+    virtual BucketSpaceStateMap& bucket_space_states() noexcept = 0;
     virtual const DistributorConfiguration& distributor_config() const noexcept = 0;
 };