diff --git a/vpp-api/python/setup.py b/vpp-api/python/setup.py
index a2eeff3..cd7afc8 100644
--- a/vpp-api/python/setup.py
+++ b/vpp-api/python/setup.py
@@ -13,9 +13,9 @@
 # limitations under the License.
 
 try:
-    from setuptools import setup, command, Extension
+    from setuptools import setup, Extension
 except ImportError:
-    from distutils.core import setup
+    from distutils.core import setup, Extension
 
 setup (name = 'vpp_papi',
        version = '1.2',