diff --git a/pkg/kafka/endpoint_manager.go b/pkg/kafka/endpoint_manager.go
index 4c13c76..1258382 100644
--- a/pkg/kafka/endpoint_manager.go
+++ b/pkg/kafka/endpoint_manager.go
@@ -56,9 +56,9 @@
 	// devices owned by that service need to be reconciled
 	IsDeviceOwnedByService(deviceID string, serviceType string, replicaNumber int32) (bool, error)
 
-	// getReplicaAssignment returns the replica number of the service that owns the deviceID.  This is used by the
+	// GetReplicaAssignment returns the replica number of the service that owns the deviceID.  This is used by the
 	// test only
-	getReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error)
+	GetReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error)
 }
 
 type service struct {
@@ -150,7 +150,7 @@
 	return m.getReplica() == ReplicaID(replicaNumber), nil
 }
 
-func (ep *endpointManager) getReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error) {
+func (ep *endpointManager) GetReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error) {
 	owner, err := ep.getOwner(deviceID, serviceType)
 	if err != nil {
 		return 0, nil