diff --git a/anonymizer_core_refactored_onnx.py b/anonymizer_core_refactored_onnx.py
index adf2b2d..c9e99bc 100644
--- a/anonymizer_core_refactored_onnx.py
+++ b/anonymizer_core_refactored_onnx.py
@@ -17,6 +17,7 @@ import json
 import logging
 import os
 import re
+import sys
 from concurrent.futures import ProcessPoolExecutor
 
 log = logging.getLogger(__name__)