From d57f1c02a9716c4b1f20bcb1c6918c1d0b46fd5d Mon Sep 17 00:00:00 2001
From: Runji Wang <wangrunji0408@163.com>
Date: Sun, 8 Oct 2023 05:49:12 +0800
Subject: [PATCH] fix(sqlsmith): recover functions for sqlsmith (#12660)

Signed-off-by: Runji Wang <wangrunji0408@163.com>
---
 src/tests/sqlsmith/src/bin/main.rs | 2 --
 src/tests/sqlsmith/src/lib.rs      | 2 ++
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tests/sqlsmith/src/bin/main.rs b/src/tests/sqlsmith/src/bin/main.rs
index 2f44031d06ab9..8e71500ec6a7f 100644
--- a/src/tests/sqlsmith/src/bin/main.rs
+++ b/src/tests/sqlsmith/src/bin/main.rs
@@ -20,8 +20,6 @@ use risingwave_sqlsmith::print_function_table;
 use risingwave_sqlsmith::runner::{generate, run, run_differential_testing};
 use tokio_postgres::NoTls;
 
-risingwave_expr_impl::enable!();
-
 #[derive(ClapParser, Debug, Clone)]
 #[clap(about, version, author)]
 struct Opt {
diff --git a/src/tests/sqlsmith/src/lib.rs b/src/tests/sqlsmith/src/lib.rs
index 2f7e1ce5eb14b..ebb0682a7aaaf 100644
--- a/src/tests/sqlsmith/src/lib.rs
+++ b/src/tests/sqlsmith/src/lib.rs
@@ -17,6 +17,8 @@
 #![feature(lazy_cell)]
 #![feature(box_patterns)]
 
+risingwave_expr_impl::enable!();
+
 use std::collections::{HashMap, HashSet};
 
 use anyhow::{bail, Result};