diff --git a/setup_tools/magicinstaller/requirements/__init__.py b/setup_tools/magicinstaller/requirements/__init__.py
index a12e3e3..1b36cc4 100644
--- a/setup_tools/magicinstaller/requirements/__init__.py
+++ b/setup_tools/magicinstaller/requirements/__init__.py
@@ -9,13 +9,13 @@
 # from .tts_package import TTS
 from .pytube_package import PyTube
 from .whisper_package import Whisper
-from setup_tools.magicinstaller.requirement import SimpleRequirementInit, CompareAction, SimpleRequirement
+from setup_tools.magicinstaller.requirement import SimpleRequirementInit, CompareAction
 
 requirements = [
     Packaging(),  # Allows for version checks
 
     # TTS(),
-    SimpleRequirement('wheel'),
+    SimpleRequirementInit('wheel'),
 
     # SimpleRequirementInit('numpy', CompareAction.EQ, '1.23.5'),
     NoColabRequirement('numpy', CompareAction.EQ, '1.23.5'),  # Don't install this one when in google colab