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