diff --git a/sources/sql/src/producer.rs b/sources/sql/src/producer.rs
index 5c9f365..0c4ef45 100644
--- a/sources/sql/src/producer.rs
+++ b/sources/sql/src/producer.rs
@@ -632,14 +632,10 @@ mod tests {
         ];
 
         for (query, expected) in tests {
-            // let dialect = GenericDialect {};
-            // let orig_ast = Parser::parse_sql(&dialect, query).unwrap();
-            // println!("{}", orig_ast[0]);
+            let dialect: Arc<dyn Dialect> = Arc::new(GenericDialect {});
             let plan = ctx.sql(query).await.unwrap().into_unoptimized_plan();
-            // println!("{:?}", plan);
 
-            let ast = query_to_sql(&plan);
-            // println!("{:?}", ast);
+            let ast = query_to_sql(&plan, dialect);
 
             assert!(ast.is_ok());
             let actual = format!("{}", ast.unwrap());