diff --git a/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithFlush.cshtml b/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithFlush.cshtml
index 646d08f7bb..c3be440959 100644
--- a/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithFlush.cshtml
+++ b/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithFlush.cshtml
@@ -6,3 +6,6 @@
 }
 @RenderBody()
 @await RenderSectionAsync("content")
+@{
+    WaitService.NotifyClient();
+}
\ No newline at end of file
diff --git a/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithPartialAndFlush.cshtml b/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithPartialAndFlush.cshtml
index 8dfc88f054..1c6110e38d 100644
--- a/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithPartialAndFlush.cshtml
+++ b/test/WebSites/RazorWebSite/Views/Shared/_LayoutWithPartialAndFlush.cshtml
@@ -8,3 +8,6 @@
 @await Html.PartialAsync("_PartialThatSetsTitle")
 @await RenderSectionAsync("content")
 More content from layout
+@{
+    WaitService.NotifyClient();
+}