fixed a test

Rippling · Mar 14, 2024 · 6abd221 · 6abd221
1 parent 59327f0
commit 6abd221
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flux_sdk/pension/utils/tests/test_ascensus/test_ascensus_180.py b/flux_sdk/pension/utils/tests/test_ascensus/test_ascensus_180.py
@@ -1,6 +1,6 @@
 import os
 import unittest
-from datetime import datetime
+from datetime import date, datetime
 from decimal import Decimal
 
 from flux_sdk.flux_core.data_models import (
@@ -81,7 +81,7 @@ def setUp(self) -> None:
         employee.dob = datetime(1990, 1, 1)
         employee.start_date = datetime(2020, 1, 1)
         employee.original_hire_date = datetime(2020, 1, 1)
-        employee.termination_date = datetime(2020, 6, 1)
+        employee.termination_date = date(2020, 6, 1)
         employee.start_date = datetime(2021, 1, 1)
         employee.status = EmployeeState.ACTIVE
         employee.is_international_employee = False