diff --git a/modelscan/modelscan.py b/modelscan/modelscan.py
index 5ea44fd..9d100d6 100644
--- a/modelscan/modelscan.py
+++ b/modelscan/modelscan.py
@@ -4,7 +4,7 @@
 from modelscan.settings import DEFAULT_SETTINGS
 
 from pathlib import Path
-from typing import List, Union, Dict, Any
+from typing import List, Union, Dict, Any, Optional
 from datetime import datetime
 
 from modelscan.error import ModelScanError, ErrorCategories