1 files changed, 2 insertions, 2 deletions

diff --git a/day2/__init__.py b/day2/__init__.py
index 988c32e..1d65d13 100644
--- a/day2/__init__.py
+++ b/day2/__init__.py
@@ -2,7 +2,7 @@ import re
 from dataclasses import dataclass
 from typing import Generator
 
-from aoc import AssignmentBase
+from aoc import BaseAssignment
 
 
 matcher = re.compile(
@@ -18,7 +18,7 @@ class Item:
     password: str
 
 
-class Assignment(AssignmentBase):
+class Assignment(BaseAssignment):
     def parse_item(self, item: str) -> Item:
         match = matcher.match(item).groupdict()
         return Item(