1 files changed, 14 insertions, 8 deletions

diff --git a/aoc/__main__.py b/aoc/__main__.py
index f5f08d3..596abee 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import enum
 import typing
 import importlib
@@ -25,15 +26,17 @@ def kwargs(kwarg: str) -> List:
 
 
 @app.command()
-def run(day: str,
-         part: str = typer.Option("One", '--part', help="Assignment part. Defaults to 'One'.", show_choices=True),
-         example: bool = typer.Option(False, '--example', help="Use an example input file"),
-         kwargs: List[str] = typer.Argument(None)
-         ):
-
+def run(
+    day: str,
+    part: str = typer.Option(
+        "1", "--part", help="Assignment part. Defaults to 'One'.", show_choices=True
+    ),
+    example: bool = typer.Option(False, "--example", help="Use an example input file"),
+    kwargs: List[str] = typer.Argument(None),
+):
     assignment_day = importlib.import_module(day)
 
-    Assignment = getattr(assignment_day, f"Assignment{part}")
+    Assignment = getattr(assignment_day, f"Assignment{AssignmentPart(part).name}")
     assignment = Assignment(
         path=os.path.dirname(assignment_day.__file__), **dict(kwargs)
     )
@@ -46,7 +49,10 @@ def run(day: str,
     )
     end = perf_counter()
     delta = end - start
-    typer.secho(f'\n{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}', fg="green")
+    typer.secho(
+        f'\n{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}',
+        fg="green",
+    )
 
 
 @app.command()