diff --git a/python/setup.py.in b/python/setup.py.in
index b376be0ea373f089ef17f27435d979712fbdff72..ee19294ad5c884cf73a4f14290f61f0b345ea8c7 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -27,7 +27,7 @@ def _get_version_detail(idx):
     if re.match('@TAG_VERSION_REGEX@', '@PADDLE_VERSION@'):
         version_details = '@PADDLE_VERSION@'.split('.')
 
-        if len(version_details) == 3:
+        if len(version_details) >= 3:
             return version_details[idx]
 
     return 0