diff --git a/sbb_binarize/cli.py b/sbb_binarize/cli.py
index ddfbde6..16145bb 100644
--- a/sbb_binarize/cli.py
+++ b/sbb_binarize/cli.py
@@ -6,7 +6,7 @@ from click import command, option, argument, version_option, types
 from .sbb_binarize import SbbBinarizer
 
 @command()
-@version_option()
+@version_option(package_name="sbb-binarization")
 @option('--model-dir', '-m', type=types.Path(exists=True, file_okay=False), required=True, help='directory containing models for prediction')
 @argument('input_image')
 @argument('output_image')
diff --git a/setup.py b/setup.py
index 7ab6e02..437730c 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@ setup(
     author='Vahid Rezanezhad',
     url='https://github.com/qurator-spk/sbb_binarization',
     license='Apache License 2.0',
-    packages=find_packages(exclude=('tests', 'docs')),
+    packages=find_packages(exclude=('test', 'repo')),
     include_package_data=True,
     package_data={'': ['*.json', '*.yml', '*.yaml']},
     install_requires=install_requires,