2 files changed, 2 insertions, 2 deletions

diff --git a/aoc/__init__.py b/aoc/__init__.py
index b07ead1..70c3ac0 100644
--- a/aoc/__init__.py
+++ b/aoc/__init__.py
@@ -4,7 +4,7 @@ from collections import Iterator
 from typing import Generator, Any
 
 
-class AssignmentBase(ABC):
+class BaseAssignment(ABC):
     def __init__(self, path):
         self.path = path
 
diff --git a/aoc/__main__.py b/aoc/__main__.py
index da0496e..8c6a21e 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -26,4 +26,4 @@ if __name__ == '__main__':
     Assignment = getattr(assignment_day, f'Assignment{args.part}')
     assignment = Assignment(path=os.path.dirname(assignment_day.__file__))
 
-    print(assignment.run(input=assignment.read_input()))
+    print(assignment.run(input=assignment.read_input(example=args.example)))