diff --git a/setup.py b/setup.py
index 362da1aa..e2a4b77f 100644
--- a/setup.py
+++ b/setup.py
@@ -40,7 +40,12 @@ def enable_sparse() -> bool:
 
 def is_editable_mode() -> bool:
     commands = [arg.lower() for arg in sys.argv]
-    return "develop" in commands or "--editable" in commands or "-e" in commands
+    return (
+        "develop" in commands
+        or "--editable" in commands
+        or "-e" in commands
+        or "editable_wheel" in commands
+    )
 
 
 class CMakeExtension(Extension):
@@ -60,7 +65,7 @@ def run(self):
     def build_cmake(self, ext: CMakeExtension):
         build_dir = os.path.abspath(self.build_temp)
         install_dir = os.path.abspath(self.build_lib)
-        if is_editable_mode:
+        if is_editable_mode():
             install_dir = ext.cmake_file_path
 
         cmake_args = [