diff --git a/test/integration_test/services/test_checkpoint_service_integration.py b/test/integration_test/services/test_checkpoint_service_integration.py
index de5170188..71a0758e6 100644
--- a/test/integration_test/services/test_checkpoint_service_integration.py
+++ b/test/integration_test/services/test_checkpoint_service_integration.py
@@ -23,7 +23,6 @@ unlike the unit tests which use mocks.
 
 import pytest
 from api.db.services.checkpoint_service import CheckpointService
-from api.db.db_models import TaskCheckpoint
 
 
 class TestCheckpointServiceIntegration: