diff --git a/com.unity.ml-agents/Documentation~/filter.yml b/com.unity.ml-agents/Documentation~/filter.yml
index 595840c5ea..ce144daf61 100755
--- a/com.unity.ml-agents/Documentation~/filter.yml
+++ b/com.unity.ml-agents/Documentation~/filter.yml
@@ -1,16 +1,10 @@
 apiRules:
 - exclude:
-    uidRegex: ^Unity.MLAgents\.Tests$
+    uidRegex: .*Test.*
     type: Namespace
 - exclude:
     uidRegex: ^Unity.MLAgents\.CommunicatorObjects$
     type: Namespace
-- exclude:
-    uidRegex: ^Unity.MLAgents\.Tests\.Communicator$
-    type: Namespace
-- exclude:
-    uidRegex: ^Unity.MLAgents\.Tests\.Actuators$
-    type: Namespace
 - exclude:
     uidRegex: ^Unity.MLAgents\.Editor$
     type: Namespace