diff --git a/reflex/istate/wrappers.py b/reflex/istate/wrappers.py
index d4e74cf8a..865bd6c63 100644
--- a/reflex/istate/wrappers.py
+++ b/reflex/istate/wrappers.py
@@ -6,7 +6,7 @@ from reflex.istate.proxy import ReadOnlyStateProxy
 from reflex.state import _split_substate_key, _substate_key, get_state_manager
 
 
-async def get_state(token, state_cls: Any | None = None) -> ReadOnlyStateProxy:
+async def get_state(token: str, state_cls: Any | None = None) -> ReadOnlyStateProxy:
     """Get the instance of a state for a token.
 
     Args:
diff --git a/reflex/utils/prerequisites.py b/reflex/utils/prerequisites.py
index 072d34a08..8165e02f6 100644
--- a/reflex/utils/prerequisites.py
+++ b/reflex/utils/prerequisites.py
@@ -1397,7 +1397,9 @@ def initialize_default_app(app_name: str):
     initialize_app_directory(app_name)
 
 
-def validate_and_create_app_using_remote_template(app_name, template, templates):
+def validate_and_create_app_using_remote_template(
+    app_name: str, template: str, templates: dict[str, Template]
+):
     """Validate and create an app using a remote template.
 
     Args: