diff --git a/sampler_scheduler_metrics_tester.py b/sampler_scheduler_metrics_tester.py
index 97a7f8a..c949cf7 100644
--- a/sampler_scheduler_metrics_tester.py
+++ b/sampler_scheduler_metrics_tester.py
@@ -5,6 +5,7 @@ import time
 import comfy.samplers
 import comfy.sample
 import comfy.utils
+import os
 import re
 import cv2 # For Laplacian and Gradient calculations