diff --git a/setup.py b/setup.py
index 97e462f..7960bfb 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
     packages=find_packages(),
     entry_points={
         "console_scripts": [
-            "sfmetadataextractor=sfmetadataextractor.cli:main",
+            "sfmetadataextractor=sfmetadataextractor.cli:app",
         ],
     },
     install_requires=[
diff --git a/sfmetadataextractor/cli.py b/sfmetadataextractor/cli.py
index 557899e..960660f 100644
--- a/sfmetadataextractor/cli.py
+++ b/sfmetadataextractor/cli.py
@@ -108,6 +108,7 @@ def generate_unit_test(
             "-o",
             help="The output unit test class file."),
 ):
+    """Generate unit test class for the input apex class."""
     typer.echo(f"Generating unit tests for {input_file}")
     generate_tests_handler = generate_tests.generateTests(input_file, output_file)
     generate_tests_handler.generate()