diff --git a/src/tools/trainer.py b/src/tools/trainer.py
index a0bb35b..25de243 100644
--- a/src/tools/trainer.py
+++ b/src/tools/trainer.py
@@ -336,6 +336,7 @@ def on_train_epoch_end(self, trainer, pl_module):
         ) else
         args.vae_emb[1] if
         (
+            args.vae_emb != None and
             len(args.vae_emb) > 1 and VAE_MODE == 'train' and os.path.isfile(
                 args.vae_emb[1])
         ) else