diff --git a/test/Core.Test/NotificationHub/NotificationHubPushNotificationServiceTests.cs b/test/Core.Test/NotificationHub/NotificationHubPushNotificationServiceTests.cs
index 831d048224..2ae3d739a8 100644
--- a/test/Core.Test/NotificationHub/NotificationHubPushNotificationServiceTests.cs
+++ b/test/Core.Test/NotificationHub/NotificationHubPushNotificationServiceTests.cs
@@ -507,6 +507,7 @@ public class NotificationHubPushNotificationServiceTests
             UserId = notification.UserId,
             OrganizationId = notification.OrganizationId,
             InstallationId = installationId,
+            TaskId = notification.TaskId,
             Title = notification.Title,
             Body = notification.Body,
             CreationDate = notification.CreationDate,