diff --git a/bin/lsb-set b/bin/lsb-set
index 5340399..6823a55 100755
--- a/bin/lsb-set
+++ b/bin/lsb-set
@@ -72,7 +72,8 @@ parser.set_defaults(input_image_file = './pictures/Lenna.png',
 try:
     generator = getattr(generators, options.generator_function)()
 except AttributeError as e:
-    raise e
+    print("Unknown generator: {}".format(options.generator_function))
+    exit(1)
 
 if options.hide:
     if options.secret_message != "" and options.secret_file == "":
diff --git a/stegano/lsbset/lsbset.py b/stegano/lsbset/lsbset.py
index e39388a..f7f24dc 100644
--- a/stegano/lsbset/lsbset.py
+++ b/stegano/lsbset/lsbset.py
@@ -174,7 +174,8 @@ if __name__ == '__main__':
     try:
         generator = getattr(generators, options.generator_function)()
     except AttributeError as e:
-        raise e
+        print("Unknown generator: {}".format(options.generator_function))
+        exit(1)
 
     if options.hide:
         if options.secret_message != "" and options.secret_file == "":