diff --git a/src/cdk/v2/destinations/koala/procWorkflow.yaml b/src/cdk/v2/destinations/koala/procWorkflow.yaml
index 9ec0202b13..dc245f61c4 100644
--- a/src/cdk/v2/destinations/koala/procWorkflow.yaml
+++ b/src/cdk/v2/destinations/koala/procWorkflow.yaml
@@ -29,7 +29,8 @@ steps:
         identifies: [{
           type: $.context.messageType,
           sent_at: .message.().({{{{$.getGenericPaths("timestamp")}}}}),
-          traits: koTraits 
+          traits: koTraits,
+          source: "rudderstack-cloud"
         }]
       };
 
diff --git a/test/integrations/destinations/koala/processor/data.ts b/test/integrations/destinations/koala/processor/data.ts
index b866353066..ec1fcd3477 100644
--- a/test/integrations/destinations/koala/processor/data.ts
+++ b/test/integrations/destinations/koala/processor/data.ts
@@ -159,6 +159,7 @@ export const data = [
                     {
                       type: 'identify',
                       sent_at: '2024-01-23T08:35:17.342Z',
+                      source: 'rudderstack-cloud',
                       traits: {
                         FirstName: 'John',
                         LastName: 'Doe',