diff --git a/src/main/java/org/openrewrite/java/testing/assertj/AdoptAssertJDurationAssertions.java b/src/main/java/org/openrewrite/java/testing/assertj/AdoptAssertJDurationAssertions.java index dd9a71640..ef26a84bd 100644 --- a/src/main/java/org/openrewrite/java/testing/assertj/AdoptAssertJDurationAssertions.java +++ b/src/main/java/org/openrewrite/java/testing/assertj/AdoptAssertJDurationAssertions.java @@ -129,10 +129,11 @@ private J.MethodInvocation simplifyMultipleAssertions(J.MethodInvocation m, Exec return m; } - J.MethodInvocation assertThatArg = (J.MethodInvocation) ((J.MethodInvocation) select).getArguments().get(0); - if (!(assertThatArg instanceof J.MethodInvocation)) { + Expression assertThatArgumentExpr = ((J.MethodInvocation) select).getArguments().get(0); + if (!(assertThatArgumentExpr instanceof J.MethodInvocation)) { return m; } + J.MethodInvocation assertThatArg = (J.MethodInvocation) assertThatArgumentExpr; if (isZero(isEqualToArg) && checkIfRelatedToDuration(assertThatArg)) { String formatted_template = formatTemplate("assertThat(#{any()}).%s();", m.getSimpleName(), asDescription);