diff --git a/.DS_Store b/.DS_Store
index af22e28..82e3a1d 100644
Binary files a/.DS_Store and b/.DS_Store differ
diff --git a/trt_utilities.py b/trt_utilities.py
index 77999e9..b12b0ac 100644
--- a/trt_utilities.py
+++ b/trt_utilities.py
@@ -462,4 +462,4 @@ class Engine:
             return self.tensors
         
         # Use safe wrapper with CUDA graph fallback for cloud environments
-        return safe_cuda_call_with_graph_fallback(_do_inference, use_cuda_graph=use_cuda_graph)
+        return safe_cuda_call_with_graph_fallback(_do_inference)