diff --git a/tests/test_device.py b/tests/test_device.py
index 2385389..f7dbb6e 100755
--- a/tests/test_device.py
+++ b/tests/test_device.py
@@ -468,7 +468,7 @@ def __copy__(self):
     def test_default_noise_model(self, requires_api):
         """Test that the default noise model is correctly set."""
         dev = SimulatorDevice(wires=(0,))
-        assert dev.noise_model == "ideal"
+        assert dev.job["input"]["noise"]["model"] == "ideal"
 
 class TestJobAttribute:
     """Tests job creation with mocked submission."""
@@ -731,4 +731,4 @@ def mock_submit_job(*args):
 
         dev.apply(tape.operations)
 
-        assert dev.job["noise"]["model"] == "forte-1"
+        assert dev.job["input"]["noise"]["model"] == "forte-1"