diff --git a/samples/python/detectron2/infer.py b/samples/python/detectron2/infer.py
index 45edaf3c..f4cba866 100644
--- a/samples/python/detectron2/infer.py
+++ b/samples/python/detectron2/infer.py
@@ -120,7 +120,7 @@ def infer(self, batch, scales=None, nms_threshold=None):
             self.inputs[0]["allocation"], np.ascontiguousarray(batch)
         )
 
-        self.context.execute_v2(self.allocations)
+        self.context.execute_v2([d.device_ptr for d in self.device_memories])
         for o in range(len(outputs)):
             common.memcpy_device_to_host(outputs[o], self.outputs[o]["allocation"])