diff --git a/tests/test_on_df.py b/tests/test_on_df.py
index a3a1a82ccd8..ac5f6f8bc23 100644
--- a/tests/test_on_df.py
+++ b/tests/test_on_df.py
@@ -1,10 +1,10 @@
 import atexit
 import io
 import os.path
-import sys
 import time
 import unittest
 
+import pandas as pd
 from chdb.dataframe import Table, pandas_read_parquet
 from utils import current_dir
 
@@ -40,6 +40,8 @@
 # run print at exit
 atexit.register(lambda: print("\n" + output.getvalue()))
 
+pd.set_option("display.max_columns", 10)
+pd.set_option("display.width", 200)
 
 class TestRunOnDf(unittest.TestCase):