diff --git a/tests/pytest/client/version.py b/tests/pytest/client/version.py
index 7cbeeb60df54e8d89fdcc7815a2b2757793dfaec..ca6ce69780159bd819655c9c6a0b46de5e07bc70 100644
--- a/tests/pytest/client/version.py
+++ b/tests/pytest/client/version.py
@@ -15,6 +15,7 @@ import sys
 from util.log import *
 from util.cases import *
 from util.sql import *
+from math import floor
 
 
 class TDTestCase:
@@ -37,13 +38,12 @@ class TDTestCase:
 
         sql = "select client_version()"
         ret = tdSql.query(sql)
-        version = tdSql.getData(0, 0)[0:3]        
-        expectedVersion_dev = "2.0"
-        expectedVersion_master = "2.1"
-        if(version == expectedVersion_dev or version == expectedVersion_master):
+        version = floor(float(tdSql.getData(0, 0)[0:3]))        
+        expectedVersion = 2
+        if(version == expectedVersion):
             tdLog.info("sql:%s, row:%d col:%d data:%s == expect" % (sql, 0, 0, version))
         else:
-            tdLog.exit("sql:%s, row:%d col:%d data:%s != expect:%s or %s " % (sql, 0, 0, version, expectedVersion_dev, expectedVersion_master))
+            tdLog.exit("sql:%s, row:%d col:%d data:%d != expect:%d " % (sql, 0, 0, version, expectedVersion))
       
 
     def stop(self):