diff --git a/optimum/commands/optimum_cli.py b/optimum/commands/optimum_cli.py
index 64a7075c6c..460e8a1d59 100644
--- a/optimum/commands/optimum_cli.py
+++ b/optimum/commands/optimum_cli.py
@@ -15,6 +15,7 @@
 
 import importlib
 from pathlib import Path
+from sys import exit
 from typing import Dict, List, Optional, Tuple, Type, Union
 
 from ..subpackages import load_subpackages