diff --git a/reflex/vars/function.py b/reflex/vars/function.py
index f600af5e0..bc619f18d 100644
--- a/reflex/vars/function.py
+++ b/reflex/vars/function.py
@@ -4,15 +4,10 @@ from __future__ import annotations
 
 import dataclasses
 import sys
-
-from typing_extensions import (
+from typing import (
     Any,
     Callable,
-    Concatenate,
-    Generic,
     Optional,
-    ParamSpec,
-    Protocol,
     Sequence,
     Tuple,
     Type,
@@ -21,6 +16,8 @@ from typing_extensions import (
     overload,
 )
 
+from typing_extensions import Concatenate, Generic, ParamSpec, Protocol
+
 from reflex.utils import format
 from reflex.utils.types import GenericType