diff --git a/python/mlx/extension.py b/python/mlx/extension.py
index afece6d83..ecf3c52e6 100644
--- a/python/mlx/extension.py
+++ b/python/mlx/extension.py
@@ -6,7 +6,7 @@ import subprocess
 import sys
 from pathlib import Path
 
-from setuptools import Extension, find_namespace_packages, setup
+from setuptools import Extension
 from setuptools.command.build_ext import build_ext
 
 import mlx