diff --git a/server/src/main/scala/za/co/absa/atum/server/api/controller/CheckpointController.scala b/server/src/main/scala/za/co/absa/atum/server/api/controller/CheckpointController.scala
index 8b6121a05..402b67884 100644
--- a/server/src/main/scala/za/co/absa/atum/server/api/controller/CheckpointController.scala
+++ b/server/src/main/scala/za/co/absa/atum/server/api/controller/CheckpointController.scala
@@ -24,6 +24,11 @@ import zio.macros.accessible
 
 @accessible
 trait CheckpointController {
-  def createCheckpointV2(checkpointDTO: CheckpointDTO): IO[ErrorResponse, SingleSuccessResponse[CheckpointDTO]]
+
   def createCheckpointV1(checkpointDTO: CheckpointDTO): IO[ErrorResponse, CheckpointDTO]
+
+  def createCheckpointV2(
+    checkpointDTO: CheckpointDTO
+  ): IO[ErrorResponse, SingleSuccessResponse[CheckpointDTO]]
+
 }
diff --git a/server/src/test/scala/za/co/absa/atum/server/api/controller/PartitioningControllerIntegrationTests.scala b/server/src/test/scala/za/co/absa/atum/server/api/controller/PartitioningControllerIntegrationTests.scala
index ed6284296..635b0f126 100644
--- a/server/src/test/scala/za/co/absa/atum/server/api/controller/PartitioningControllerIntegrationTests.scala
+++ b/server/src/test/scala/za/co/absa/atum/server/api/controller/PartitioningControllerIntegrationTests.scala
@@ -58,9 +58,11 @@ object PartitioningControllerIntegrationTests extends ZIOSpecDefault with TestDa
       ),
       suite("CreateOrUpdateAdditionalDataSuite")(
         test("Returns expected AdditionalDataSubmitDTO") {
-          assertZIO(PartitioningController.createOrUpdateAdditionalData(additionalDataSubmitDTO1))(
-            equalTo(SingleSuccessResponse(additionalDataSubmitDTO1))
-          )
+          for {
+            result <- PartitioningController.createOrUpdateAdditionalData(additionalDataSubmitDTO1)
+            expected = SingleSuccessResponse(additionalDataSubmitDTO1, uuid)
+            actual = result.copy(requestId = uuid)
+          } yield assert(actual)(equalTo(expected))
         },
         test("Returns expected InternalServerErrorResponse") {
           assertZIO(PartitioningController.createOrUpdateAdditionalData(additionalDataSubmitDTO2).exit)(