diff --git a/extraction/core/extractor.py b/extraction/core/extractor.py
index a050bbfc23141ef5fa9b59abd9f1c2f3fe881e15..9f769eaafe6fa6f97f53721a52761f4f74e158da 100644
--- a/extraction/core/extractor.py
+++ b/extraction/core/extractor.py
@@ -1,7 +1,7 @@
 import logging
 from time import perf_counter
 
-from typing import List, Dict, Callable
+from typing import Union, List, Dict, Callable
 
 import numpy as np
 import pandas as pd