diff --git a/protocol-testing/Cargo.lock b/protocol-testing/Cargo.lock
index 732b4cf..b2a8f31 100644
--- a/protocol-testing/Cargo.lock
+++ b/protocol-testing/Cargo.lock
@@ -7821,8 +7821,7 @@ dependencies = [
 
 [[package]]
 name = "tycho-simulation"
-version = "0.156.0"
-source = "git+https://github.com/propeller-heads/tycho-simulation.git?tag=0.156.0#1983a787440e8ae757626d808a6e619baffc52f2"
+version = "0.155.2"
 dependencies = [
  "alloy",
  "async-stream",
diff --git a/protocol-testing/src/test_runner.rs b/protocol-testing/src/test_runner.rs
index e67fa11..75bb944 100644
--- a/protocol-testing/src/test_runner.rs
+++ b/protocol-testing/src/test_runner.rs
@@ -336,11 +336,6 @@ fn validate_state(
         decoder_context,
     );
 
-    // NOTE: Once tycho-simulation is updated, you can use the new API like this:
-    // use tycho_simulation::protocol::models::DecoderContext;
-    // let context = DecoderContext::new().vm_adapter_path(adapter_contract_path_str);
-    // decoder.register_decoder_with_context::<EVMPoolState<PreCachedDB>>("test_protocol", context);
-
     // Mock a stream message, with only a Snapshot and no deltas
     let mut states: HashMap<String, ComponentWithState> = HashMap::new();
     for (id, component) in components_by_id {