diff --git a/luminance_stack_processor.py b/luminance_stack_processor.py
index 3ee6f6c..5c0c3f6 100644
--- a/luminance_stack_processor.py
+++ b/luminance_stack_processor.py
@@ -12,6 +12,8 @@ import torch
 import cv2
 from typing import Tuple, List
 import logging
+import os
+
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)