diff --git a/pom.xml b/pom.xml
index 51c897b4452f..296450231870 100644
--- a/pom.xml
+++ b/pom.xml
@@ -2450,7 +2450,14 @@
                   <reason>Use junit5 instead</reason>
                   <bannedImports>
                     <bannedImport>junit.framework.**</bannedImport>
+                    <bannedImport>org.junit.**</bannedImport>
+                    <bannedImport>static org.junit.Assert.**</bannedImport>
+                    <bannedImport>static org.junit.Assume.**</bannedImport>
                   </bannedImports>
+                  <allowedImports>
+                    <allowedImport>org.junit.jupiter.**</allowedImport>
+                    <allowedImport>org.junit.platform.**</allowedImport>
+                  </allowedImports>
                 </restrictImports>
               </rules>
             </configuration>