diff --git a/video_prediction_tools/postprocess/plot_ambs_forecast.py b/video_prediction_tools/postprocess/plot_ambs_forecast.py
index 172d1386e3ae3c4fbe6f59d8d2d9f03b263b0f59..83b1c8ff89eabb1e6850e638db5f2f88ff624ead 100644
--- a/video_prediction_tools/postprocess/plot_ambs_forecast.py
+++ b/video_prediction_tools/postprocess/plot_ambs_forecast.py
@@ -12,7 +12,7 @@ import numpy as np
 import pandas as pd
 import sys 
 from argparse import ArgumentParser
-
+import os
 import matplotlib
 
 matplotlib.use('Agg')