diff --git a/src/test/kotlin/nl/avisi/structurizr/site/generatr/site/model/MenuViewModelTest.kt b/src/test/kotlin/nl/avisi/structurizr/site/generatr/site/model/MenuViewModelTest.kt
index adc62157..9fa2a15f 100644
--- a/src/test/kotlin/nl/avisi/structurizr/site/generatr/site/model/MenuViewModelTest.kt
+++ b/src/test/kotlin/nl/avisi/structurizr/site/generatr/site/model/MenuViewModelTest.kt
@@ -23,6 +23,7 @@ class MenuViewModelTest : ViewModelTest() {
 
         assertThat(viewModel.generalItems).containsExactly(
             LinkViewModel(pageViewModel, "Home", HomePageViewModel.url()),
+            LinkViewModel(pageViewModel, "Search", SearchViewModel.url()),
         )
     }
 
@@ -36,7 +37,7 @@ class MenuViewModelTest : ViewModelTest() {
         val pageViewModel = createPageViewModel(generatorContext)
         val viewModel = MenuViewModel(generatorContext, pageViewModel)
 
-        assertThat(viewModel.generalItems[1]).isEqualTo(
+        assertThat(viewModel.generalItems[2]).isEqualTo(
             LinkViewModel(pageViewModel, "Software Systems", SoftwareSystemsPageViewModel.url())
         )
     }
@@ -56,7 +57,7 @@ class MenuViewModelTest : ViewModelTest() {
         val pageViewModel = createPageViewModel(generatorContext)
         val viewModel = MenuViewModel(generatorContext, pageViewModel)
 
-        assertThat(viewModel.generalItems[1]).isEqualTo(
+        assertThat(viewModel.generalItems[2]).isEqualTo(
             LinkViewModel(pageViewModel, "Decisions", WorkspaceDecisionsPageViewModel.url(), false)
         )
     }
@@ -73,7 +74,7 @@ class MenuViewModelTest : ViewModelTest() {
         val pageViewModel = createPageViewModel(generatorContext)
         val viewModel = MenuViewModel(generatorContext, pageViewModel)
 
-        assertThat(viewModel.generalItems.drop(1)).containsExactly(
+        assertThat(viewModel.generalItems.drop(2)).containsExactly(
             LinkViewModel(pageViewModel, "Doc 1", WorkspaceDocumentationSectionPageViewModel.url(section1)),
             LinkViewModel(pageViewModel, "Doc Title 2", WorkspaceDocumentationSectionPageViewModel.url(section2))
         )