diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java
index ff2c3e90..cdcf153b 100644
--- a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java
+++ b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java
@@ -303,6 +303,12 @@ public void testUpdateSession()
             DatabendSession session = connection.unwrap(DatabendConnection.class).getSession();
             Assert.assertEquals(session.getDatabase(), "test_basic_driver_2");
             Assert.assertEquals(session.getSettings().get("max_threads"), "1");
+
+            // test variable
+            connection.createStatement().execute("set variable a=1");
+            ResultSet r = connection.createStatement().executeQuery("select $a");
+            r.next();
+            assertEquals(r.getInt(1), 1);
         }
     }