diff --git a/cuda_core/tests/graph/test_graph_definition_lifetime.py b/cuda_core/tests/graph/test_graph_definition_lifetime.py
index cc3872bb82..a01ff8162f 100644
--- a/cuda_core/tests/graph/test_graph_definition_lifetime.py
+++ b/cuda_core/tests/graph/test_graph_definition_lifetime.py
@@ -443,7 +443,8 @@ class Callback:
             def __call__(self):
                 pass
 
-        Device()
+        device = Device()
+        device.set_current()
         graph = GraphDefinition()
         graph.callback(Callback())
         """