diff --git a/DocumentOperator.Tests/Integration/API/PdfAttachmentControllerTests.cs b/DocumentOperator.Tests/Integration/API/PdfAttachmentControllerTests.cs
index 2d397b0..364a400 100644
--- a/DocumentOperator.Tests/Integration/API/PdfAttachmentControllerTests.cs
+++ b/DocumentOperator.Tests/Integration/API/PdfAttachmentControllerTests.cs
@@ -1,5 +1,5 @@
-using DocumentService.API.Controllers; // For CheckPdfAttachmentsRequest DTO
 using DocumentService.Application.Common.DTOs;
+using DocumentService.Client.Models.Requests;
 using FluentAssertions;
 using Microsoft.AspNetCore.Mvc.Testing;
 using System.Net;
diff --git a/DocumentOperator.Tests/Integration/API/PdfValidationControllerTests.cs b/DocumentOperator.Tests/Integration/API/PdfValidationControllerTests.cs
index 4d92e58..e4ee38f 100644
--- a/DocumentOperator.Tests/Integration/API/PdfValidationControllerTests.cs
+++ b/DocumentOperator.Tests/Integration/API/PdfValidationControllerTests.cs
@@ -1,5 +1,5 @@
-using DocumentService.API.Controllers; // For Base64Request DTOs
 using DocumentService.Application.Common.DTOs;
+using DocumentService.Client.Models.Requests;
 using FluentAssertions;
 using Microsoft.AspNetCore.Mvc.Testing;
 using System.Net;