diff --git a/tests-integration/src/tests/instance_test.rs b/tests-integration/src/tests/instance_test.rs
index 813e3fcfb424..5f33b14c989f 100644
--- a/tests-integration/src/tests/instance_test.rs
+++ b/tests-integration/src/tests/instance_test.rs
@@ -1873,12 +1873,7 @@ async fn test_custom_storage(instance: Arc<dyn MockInstance>) {
                     a int null primary key,
                     ts timestamp time index,
                 )
-                PARTITION BY RANGE COLUMNS (a) (
-                    PARTITION r0 VALUES LESS THAN (1),
-                    PARTITION r1 VALUES LESS THAN (10),
-                    PARTITION r2 VALUES LESS THAN (100),
-                    PARTITION r3 VALUES LESS THAN (MAXVALUE),
-                )
+                PARTITION ON COLUMNS (a) ()
                 with(storage='{storage_name}')
                 "#
                 )
@@ -1927,11 +1922,8 @@ async fn test_custom_storage(instance: Arc<dyn MockInstance>) {
   TIME INDEX ("ts"),
   PRIMARY KEY ("a")
 )
-PARTITION BY RANGE COLUMNS ("a") (
-  PARTITION r0 VALUES LESS THAN (1),
-  PARTITION r1 VALUES LESS THAN (10),
-  PARTITION r2 VALUES LESS THAN (100),
-  PARTITION r3 VALUES LESS THAN (MAXVALUE)
+PARTITION ON COLUMNS ("a") (
+
 )
 ENGINE=mito
 WITH(