diff --git a/bittensor/mock/subtensor_mock.py b/bittensor/mock/subtensor_mock.py
index 30d58f22e0..5c2c3b42d6 100644
--- a/bittensor/mock/subtensor_mock.py
+++ b/bittensor/mock/subtensor_mock.py
@@ -624,7 +624,7 @@ def query_subtensor(
             state_at_block = state.get(block, None)
             while state_at_block is None and block > 0:
                 block -= 1
-                state_at_block = self.state.get(block, None)
+                state_at_block = state.get(block, None)
             if state_at_block is not None:
                 return SimpleNamespace(value=state_at_block)
 
diff --git a/tests/integration_tests/test_subtensor_integration.py b/tests/integration_tests/test_subtensor_integration.py
index e3661210bc..407dee848c 100644
--- a/tests/integration_tests/test_subtensor_integration.py
+++ b/tests/integration_tests/test_subtensor_integration.py
@@ -115,6 +115,16 @@ def test_get_current_block(self):
         block = self.subtensor.get_current_block()
         assert type(block) == int
 
+    def test_do_block_step(self):
+        self.subtensor.do_block_step()
+        block = self.subtensor.get_current_block()
+        assert type(block) == int
+
+    def test_do_block_step_query_previous_block(self):
+        self.subtensor.do_block_step()
+        block = self.subtensor.get_current_block()
+        self.subtensor.query_subtensor("NetworksAdded", block)
+
     def test_unstake(self):
         self.subtensor._do_unstake = MagicMock(return_value=True)