diff --git a/prompt_control/test_graph.py b/prompt_control/test_graph.py
index 8f7c45d..0210a93 100644
--- a/prompt_control/test_graph.py
+++ b/prompt_control/test_graph.py
@@ -17,6 +17,7 @@ def apply(cls, text, **kwargs):
 
 
 @mock.patch("prompt_control.utils.lora_name_to_file", find_file)
+@mock.patch("torch.cuda.current_device", lambda: "cpu")
 class GraphTests(unittest.TestCase):
     maxDiff = 4096