diff --git a/rust/cubesql/cubesql/src/compile/test/test_bi_workarounds.rs b/rust/cubesql/cubesql/src/compile/test/test_bi_workarounds.rs
index fc8d990339dcc..3d15c92450e78 100644
--- a/rust/cubesql/cubesql/src/compile/test/test_bi_workarounds.rs
+++ b/rust/cubesql/cubesql/src/compile/test/test_bi_workarounds.rs
@@ -1,3 +1,5 @@
+use pretty_assertions::assert_eq;
+
 use super::LogicalPlanTestUtils;
 use crate::{
     compile::{
diff --git a/rust/cubesql/cubesql/src/compile/test/test_cube_join.rs b/rust/cubesql/cubesql/src/compile/test/test_cube_join.rs
index 40ed374870b46..c9a63a8754570 100644
--- a/rust/cubesql/cubesql/src/compile/test/test_cube_join.rs
+++ b/rust/cubesql/cubesql/src/compile/test/test_cube_join.rs
@@ -1,6 +1,7 @@
 use cubeclient::models::{
     V1LoadRequestQuery, V1LoadRequestQueryFilterItem, V1LoadRequestQueryTimeDimension,
 };
+use pretty_assertions::assert_eq;
 use serde_json::json;
 
 use crate::compile::{
diff --git a/rust/cubesql/cubesql/src/compile/test/test_udfs.rs b/rust/cubesql/cubesql/src/compile/test/test_udfs.rs
index 1007857abda30..46bf3f8faeedf 100644
--- a/rust/cubesql/cubesql/src/compile/test/test_udfs.rs
+++ b/rust/cubesql/cubesql/src/compile/test/test_udfs.rs
@@ -1,3 +1,5 @@
+use pretty_assertions::assert_eq;
+
 use crate::{
     compile::{
         test::{execute_query, init_testing_logger},