diff --git a/video_prediction_tools/postprocess/plot_ambs_forecast.py b/video_prediction_tools/postprocess/plot_ambs_forecast.py
index e26f5a2b662f196fc3fe7ad7dec0134636891794..172d1386e3ae3c4fbe6f59d8d2d9f03b263b0f59 100644
--- a/video_prediction_tools/postprocess/plot_ambs_forecast.py
+++ b/video_prediction_tools/postprocess/plot_ambs_forecast.py
@@ -10,6 +10,8 @@ the AMBS-workflow)
 import xarray as xr
 import numpy as np
 import pandas as pd
+import sys 
+from argparse import ArgumentParser
 
 import matplotlib