diff --git a/src/servers/tests/py_script/mod.rs b/src/servers/tests/py_script/mod.rs
index eba376189c..3d9d9226b4 100644
--- a/src/servers/tests/py_script/mod.rs
+++ b/src/servers/tests/py_script/mod.rs
@@ -15,7 +15,8 @@
 use std::sync::Arc;
 
 use servers::error::Result;
-use servers::query_handler::{ScriptHandler, SqlQueryHandler};
+use servers::query_handler::sql::SqlQueryHandler;
+use servers::query_handler::ScriptHandler;
 use session::context::QueryContext;
 use table::test_util::MemTable;