diff --git a/inheritance-and-composition/choosing/hr.py b/inheritance-and-composition/choosing/hr.py index 804614e2b1..8e03668e25 100644 --- a/inheritance-and-composition/choosing/hr.py +++ b/inheritance-and-composition/choosing/hr.py @@ -11,7 +11,7 @@ def __init__(self): def get_policy(self, employee_id): policy = self._employee_policies.get(employee_id) if not policy: - return ValueError(employee_id) + raise ValueError(employee_id) return policy def calculate_payroll(self, employees): diff --git a/inheritance-and-composition/composition/hr.py b/inheritance-and-composition/composition/hr.py index 32e628254a..e4c976d59b 100644 --- a/inheritance-and-composition/composition/hr.py +++ b/inheritance-and-composition/composition/hr.py @@ -11,7 +11,7 @@ def __init__(self): def get_policy(self, employee_id): policy = self._employee_policies.get(employee_id) if not policy: - return ValueError(employee_id) + raise ValueError(employee_id) return policy def calculate_payroll(self, employees):