diff --git a/tests/crud/updates.rs b/tests/crud/updates.rs
index 262031ef..55af604e 100644
--- a/tests/crud/updates.rs
+++ b/tests/crud/updates.rs
@@ -1,6 +1,6 @@
 pub use super::*;
 use rust_decimal_macros::dec;
-use sea_orm::DbErr;
+use sea_orm::{query::*, DbErr};
 use uuid::Uuid;
 
 pub async fn test_update_cake(db: &DbConn) {