diff --git a/python/setup.py.in b/python/setup.py.in
index 2b052574c0716e0606e362200b1cb70cd214e2b0..db4c7f044d64980eeac4bde130c2f8ed04459fde 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -25,8 +25,9 @@ def _get_version_detail(idx):
 
     version_details = '@PADDLE_VERSION@'.split('.')
     if len(version_details) == 3:
-        if re.match('[0-9]+', version_details[idx]):
-            return int(version_details[idx])
+        detail = version_details[idx]
+        if re.match('[0-9]+', detail):
+            return int(detail)
 
     return None