diff --git a/shared/types/states/master.py b/shared/types/states/master.py
index 51659334..e1233b11 100644
--- a/shared/types/states/master.py
+++ b/shared/types/states/master.py
@@ -70,7 +70,7 @@ class MasterState(SharedState):
     cache_policy: CachePolicy[CachePolicyType]
 
 
-def get_inference_plan(
+def get_shard_assignments(
     inbox: Queue[ExternalCommand],
     outbox: Queue[ExternalCommand],
     resource_graph: ResourceGraph,