diff --git a/prompt_control/test_encode.py b/prompt_control/test_encode.py
index f1fa92a..0aa70ea 100644
--- a/prompt_control/test_encode.py
+++ b/prompt_control/test_encode.py
@@ -20,7 +20,6 @@ def run(f, *args):
         return getattr(f, f.FUNCTION)(*args)
 
 
-@mock.patch("torch.cuda.current_device", lambda: "cpu")
 class TestEncode(unittest.TestCase):
     @classmethod
     def setUpClass(cls):