diff --git a/python/setup.py.in b/python/setup.py.in
index 6732ce07cafb882ab3974346ff3de5a81aa14c97..6e439d28ac61a28e8c096f7cbdd9f225f9966a66 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 or len(version_details) == 4:
+        if len(version_details) >= 3:
             return version_details[idx]
 
     return 0