diff --git a/tests/GraphQL/ActionEngine/CompanyTaskTest.php b/tests/GraphQL/ActionEngine/CompanyTaskTest.php
index f0eb7180c..de1a7f140 100644
--- a/tests/GraphQL/ActionEngine/CompanyTaskTest.php
+++ b/tests/GraphQL/ActionEngine/CompanyTaskTest.php
@@ -77,13 +77,26 @@ public function testGetLeadTaskEngagement()
             leadTaskItems(lead_id: $lead_id) {
                 data {
                     name
-                    action {
+                    company_action {
                         name
+                        action {
+                            slug
+                        }
                     }
                     status
                     config
                     engagement_start {
                         uuid
+                        message {
+                            message
+                        }
+                        lead {
+                            title
+                        }
+                        slug
+                        entity_uuid
+                    } engagement_end {
+                        uuid
                     }
                 }
             }