diff --git a/src/cloudformation_cli_python_lib/resource.py b/src/cloudformation_cli_python_lib/resource.py
index 6017f4f3..39c2abf5 100644
--- a/src/cloudformation_cli_python_lib/resource.py
+++ b/src/cloudformation_cli_python_lib/resource.py
@@ -60,12 +60,12 @@ 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
         self._type_configuration_model_cls: Optional[
-            BaseModel
+            Type[BaseModel]
         ] = type_configuration_model_cls
         self._handlers: MutableMapping[Action, HandlerSignature] = {}
 
diff --git a/src/cloudformation_cli_python_lib/utils.py b/src/cloudformation_cli_python_lib/utils.py
index be0987a8..c5cae14a 100644
--- a/src/cloudformation_cli_python_lib/utils.py
+++ b/src/cloudformation_cli_python_lib/utils.py
@@ -140,7 +140,7 @@ class UnmodelledRequest:
     def to_modelled(
         self,
         model_cls: Type[BaseModel],
-        type_configuration_model_cls: Optional[BaseModel],
+        type_configuration_model_cls: Optional[Type[BaseModel]],
     ) -> BaseResourceHandlerRequest:
         # pylint: disable=protected-access
         return BaseResourceHandlerRequest(