diff --git a/java/gazelle/private/java/java.go b/java/gazelle/private/java/java.go
index 62ec8dc9..70d774b2 100644
--- a/java/gazelle/private/java/java.go
+++ b/java/gazelle/private/java/java.go
@@ -21,7 +21,7 @@ func IsTestPackage(pkg string) bool {
 		}
 	}
 
-	return strings.Contains(pkg, "/test")
+	return strings.Contains(pkg, "/test/") || strings.HasSuffix(pkg, "/test")
 }
 
 // This list was derived from a script along the lines of: