diff --git a/colossalai/auto_parallel/solver/sharding_strategy.py b/colossalai/auto_parallel/solver/sharding_strategy.py
index 342f6ff37..e7429d386 100644
--- a/colossalai/auto_parallel/solver/sharding_strategy.py
+++ b/colossalai/auto_parallel/solver/sharding_strategy.py
@@ -105,7 +105,7 @@ class StrategyGenerator_V2(ABC):
         self.device_mesh = device_mesh
 
     @abstractmethod
-    def generate(self, operand_mapping: Dict[str:Operand]) -> List[ShardingStrategy_V2]:
+    def generate(self, operand_mapping: Dict[str, Operand]) -> List[ShardingStrategy_V2]:
         """
         """
         pass