diff --git a/src/cloudformation_cli_python_lib/resource.py b/src/cloudformation_cli_python_lib/resource.py
index 6017f4f3..74915781 100644
--- a/src/cloudformation_cli_python_lib/resource.py
+++ b/src/cloudformation_cli_python_lib/resource.py
@@ -60,7 +60,7 @@ def __init__(
         self,
         type_name: str,
         resouce_model_cls: Type[BaseModel],
-        type_configuration_model_cls: Optional[BaseModel] = None,
+        type_configuration_model_cls: Optional[Type[BaseModel]] = None,
     ) -> None:
         self.type_name = type_name
         self._model_cls: Type[BaseModel] = resouce_model_cls