diff --git a/python/setup.py b/python/setup.py
index d9359e58b78b9f09405cff534b347a04b928a513..79f73d506d71d9771ec06625c2d0489c5190ba13 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -22,7 +22,7 @@ import platform
 
 from setuptools import find_packages
 from setuptools import setup
-
+from paddle_fl.version import fl_version
 
 def python_version():
     """