diff --git a/bob/learn/pytorch/scripts/train_drgan_multipie.py b/bob/learn/pytorch/scripts/train_drgan_multipie.py
index 70d10ceebdaab305c5c665948beaa5b7255ef615..4df5b4b1499783356d09125c00276a2c71e1b361 100644
--- a/bob/learn/pytorch/scripts/train_drgan_multipie.py
+++ b/bob/learn/pytorch/scripts/train_drgan_multipie.py
@@ -101,6 +101,7 @@ def main(user_input=None):
   if bool(args['--light']):
     if dropout:
       logger.error("The light architecture does not support dropout - so drop this option ;)")
+      sys.exit()
     from bob.learn.pytorch.architectures import DRGAN_encoder as drgan_encoder
     from bob.learn.pytorch.architectures import DRGAN_decoder as drgan_decoder
     from bob.learn.pytorch.architectures import DRGAN_discriminator as drgan_discriminator