diff --git a/src/test/kotlin/b/language/server/prob2/proBMangement/BDocumentServiceTest.kt b/src/test/kotlin/b/language/server/prob2/proBMangement/BDocumentServiceTest.kt
index 8c5fb80920597ca441b0c6df11498a061c08b783..50358868a484d33d1339600867d07c3b023291dd 100644
--- a/src/test/kotlin/b/language/server/prob2/proBMangement/BDocumentServiceTest.kt
+++ b/src/test/kotlin/b/language/server/prob2/proBMangement/BDocumentServiceTest.kt
@@ -114,14 +114,13 @@ class BDocumentServiceTest {
 
         val documentService = BDocumentService(DummyServer(), communicator, ProBKernelManager(communicator))
 
-        //Filling the internel state with some information
         documentService.checkDocument("src/test/resources/WD_M1.mch")
 
         val targetSet =  communicator.pushedDiagnostics.entries.first().value.map { value -> value.source }.toSet()
 
 
         assertEquals(1, communicator.pushedDiagnostics.entries.size)
-        assertEquals(2, communicator.pushedDiagnostics.entries.first().value.size)
+        assertEquals(3, communicator.pushedDiagnostics.entries.first().value.size)
         assertTrue(targetSet.first().contains("b-language-server/src/test/resources/WD_M1.mch"))
 
     }
@@ -134,7 +133,6 @@ class BDocumentServiceTest {
 
         val documentService = BDocumentService(DummyServer(), communicator, ProBKernelManager(communicator))
 
-        //Filling the internel state with some information
         documentService.checkDocument("src/test/resources/WD_M2.mch")
 
         communicator.pushedDiagnostics.clear()
@@ -161,7 +159,6 @@ class BDocumentServiceTest {
 
         val documentService = BDocumentService(DummyServer(), communicator, DummyProBKernelManager(communicator))
 
-        //Filling the internel state with some information
         documentService.checkDocument("src/test/resources/WD_M2.mch")
 
         documentService.checkDocument("src/test/resources/WD_M2.mch")