diff --git a/api/tests/unit_tests/services/test_document_service_save_document_with_dataset_id.py b/api/tests/unit_tests/services/test_document_service_save_document_with_dataset_id.py
index 1832e8699f..77112dfaa7 100644
--- a/api/tests/unit_tests/services/test_document_service_save_document_with_dataset_id.py
+++ b/api/tests/unit_tests/services/test_document_service_save_document_with_dataset_id.py
@@ -3,7 +3,6 @@ from unittest.mock import Mock, patch
 
 import pytest
 
-from models import DatasetProcessRule
 from models.account import Account
 from models.dataset import Dataset, Document
 from services.dataset_service import DocumentService