diff --git a/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitContains.java b/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitContains.java index ae8daa383..568903d3a 100644 --- a/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitContains.java +++ b/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitContains.java @@ -72,9 +72,9 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation m, ExecutionC if (!isTrue && !IS_FALSE.matches(method)) { return method; } - + if (!(method.getSelect() instanceof J.MethodInvocation)) { - return method; + return method; } if (!ASSERT_THAT.matches((J.MethodInvocation)method.getSelect())) { diff --git a/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitSize.java b/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitSize.java index eb9a1e30e..9ffc207d5 100644 --- a/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitSize.java +++ b/src/main/java/org/openrewrite/java/testing/assertj/UseExplicitSize.java @@ -60,11 +60,15 @@ public static class UseExplicitSizeVisitor extends JavaIsoVisitor