diff --git a/tests/test_engine.py b/tests/test_engine.py
index 6586dcf4..784ffcb9 100644
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@@ -31,7 +31,6 @@ class MockModel:
     def __init__(self, vocab_size=262):  # 256 bytes + 6 special tokens
         self.vocab_size = vocab_size
         self.config = MockConfig()
-        self._device = "cpu"
         self._device = torch.device("cpu")
 
     def get_device(self):