diff --git a/weave/legacy/weave/ref_base.py b/weave/legacy/weave/ref_base.py
index 6951fe602b3..ba92ea13ca5 100644
--- a/weave/legacy/weave/ref_base.py
+++ b/weave/legacy/weave/ref_base.py
@@ -17,7 +17,7 @@
 
 if typing.TYPE_CHECKING:
     from weave.legacy.weave import weave_types as types
-    from ..trace import weave_client
+    from ...trace import weave_client
 
 
 def _map_to_ref_strs(obj: typing.Any) -> typing.Any:
diff --git a/weave/trace/box.py b/weave/trace/box.py
index a185c50a822..d0b7db6db9d 100644
--- a/weave/trace/box.py
+++ b/weave/trace/box.py
@@ -10,7 +10,7 @@
 
 import numpy as np
 
-from weave.legacy.weave.ref_base import Ref
+from weave.trace.refs import Ref
 
 T = TypeVar("T")
 
diff --git a/weave/trace/op_type.py b/weave/trace/op_type.py
index fd948b8a2e0..c570e60436c 100644
--- a/weave/trace/op_type.py
+++ b/weave/trace/op_type.py
@@ -19,11 +19,11 @@
     get_class_source,
     is_running_interactively,
 )
+from weave.trace.op import Op
 from weave.trace.refs import ObjectRef
 
 from ..legacy.weave import environment
 from . import serializer
-from .op import Op
 
 WEAVE_OP_PATTERN = re.compile(r"@weave\.op(\(\))?")
 WEAVE_OP_NO_PAREN_PATTERN = re.compile(r"@weave\.op(?!\()")