diff --git a/ai-compliance-sdk/internal/api/handlers/rag_handlers_test.go b/ai-compliance-sdk/internal/api/handlers/rag_handlers_test.go
index 8000611..a1f86e6 100644
--- a/ai-compliance-sdk/internal/api/handlers/rag_handlers_test.go
+++ b/ai-compliance-sdk/internal/api/handlers/rag_handlers_test.go
@@ -13,11 +13,14 @@ import (
 func TestAllowedCollections(t *testing.T) {
 	allowed := []string{
 		"bp_compliance_ce",
-		"bp_compliance_recht",
 		"bp_compliance_gesetze",
 		"bp_compliance_datenschutz",
+		"bp_compliance_gdpr",
 		"bp_dsfa_corpus",
+		"bp_dsfa_templates",
+		"bp_dsfa_risks",
 		"bp_legal_templates",
+		"bp_iace_libraries",
 	}
 
 	for _, c := range allowed {