diff --git a/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py b/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py
index d26f083613..be01a37cd1 100644
--- a/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py
+++ b/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py
@@ -768,7 +768,7 @@ def recursively_remove_trigger(obj: Any) -> Any:
         for i, v in enumerate(obj):
             obj[i] = recursively_remove_trigger(v)
     else:
-        if not hasattr(obj, "__dict__") or isinstance(obj, (type,)):
+        if not hasattr(obj, "__dict__") or isinstance(obj, (type, torch.Tensor)):
             return obj
         for k, v in obj.__dict__.items():
             if k[:2] != "__" or k[-2:] != "__":