diff --git a/CMakeLists.txt b/CMakeLists.txt
index a3e62bc26750..d8d23f90353d 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -693,7 +693,7 @@ if(NOT DEFINED ENV{CONDA_BUILD})
   message(STATUS ${CMAKE_CURRENT_BINARY_DIR})
   add_custom_target(
     tvm_cython ALL
-    ${Python_EXECUTABLE} -I setup.py build_ext --inplace
+    ${Python_EXECUTABLE} setup.py build_ext --inplace
     WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/python
   )
   add_dependencies(tvm_cython tvm)
diff --git a/python/setup.py b/python/setup.py
index cf2eff2a3af4..679f5078d3c1 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -20,6 +20,7 @@
 import pathlib
 import shutil
 import sys
+import sys
 
 from setuptools import find_packages
 from setuptools.dist import Distribution