diff --git a/packages/python-sdk/e2b/sandbox/utils.py b/packages/python-sdk/e2b/sandbox/utils.py
index daf9d4a8b..e147409e5 100644
--- a/packages/python-sdk/e2b/sandbox/utils.py
+++ b/packages/python-sdk/e2b/sandbox/utils.py
@@ -1,10 +1,10 @@
-from typing import TypeVar, Any, cast, Optional, Type
+from typing import TypeVar, Any, Generic, cast, Optional, Type
 import functools
 
 T = TypeVar("T")
 
 
-class class_method_variant(object):
+class class_method_variant(Generic[T]):
     def __init__(self, class_method_name):
         self.class_method_name = class_method_name