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