diff --git a/setup2.py b/setup2.py
index dbf4d43..88e65c0 100644
--- a/setup2.py
+++ b/setup2.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-from setuptools import setup
+from setuptools import setup, Extension
 from Cython.Build import cythonize
 from Cython.Distutils import build_ext