diff --git a/setup.py b/setup.py
index d7b5829..c4987d1 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,7 @@
 import os
-from distutils.core import setup
 from typing import List
 
-from setuptools import find_packages
+from setuptools import find_packages, setup
 
 
 def find_stub_files(name: str) -> List[str]: