diff --git a/src/runtime2/branch.rs b/src/runtime2/branch.rs
index 80a2a191b1a8f0a8eaed8f68b46fd08309d1bd2d..1b296960bcfabfc69b0176b84cd9b13b012e6bd3 100644
--- a/src/runtime2/branch.rs
+++ b/src/runtime2/branch.rs
@@ -3,7 +3,8 @@ use std::ops::{Index, IndexMut};
 
 use crate::protocol::ComponentState;
 use crate::protocol::eval::{Value, ValueGroup};
-use crate::runtime2::port::PortIdLocal;
+
+use super::port::PortIdLocal;
 
 /// Generic branch ID. A component will always have one branch: the
 /// non-speculative branch. This branch has ID 0. Hence in a speculative context