diff --git a/scratch/johannes/gail_options_image.py b/scratch/johannes/gail_options_image.py
index ac1676a..03849d5 100644
--- a/scratch/johannes/gail_options_image.py
+++ b/scratch/johannes/gail_options_image.py
@@ -24,6 +24,7 @@ from tqdm import tqdm
 from src.policies.options import OptionsCnnPolicy
 from src.gail.options import OptionsEnv, LLOptions, HLOptions, RenderOptions
 from src.gail.train import train_discriminator, train_generator
+from src.evaluation.evaluation import Evaluation
 
 model_name = 'gail_options_image'
 Env = NRasterizedRandomAgent