diff --git a/setup_wizard.py b/setup_wizard.py
index 0597abd2..e6ed5cfb 100644
--- a/setup_wizard.py
+++ b/setup_wizard.py
@@ -9,6 +9,7 @@ Usage:
 """
 from __future__ import annotations
 
+import argparse
 import json
 import os
 import platform