polish

openrewrite · Oct 18, 2023 · cd996a0 · cd996a0
1 parent 38acaab
commit cd996a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/openrewrite/java/testing/jmockit/JMockitExpectationsToMockitoWhen.java b/src/main/java/org/openrewrite/java/testing/jmockit/JMockitExpectationsToMockitoWhen.java
@@ -89,7 +89,7 @@ public J visitNewClass(J.NewClass newClass, ExecutionContext executionContext) {
       maybeAddImport("org.mockito.Mockito", "when");
       maybeRemoveImport("mockit.Expectations");
 
-      return newMethod.withPrefix(newClass.getPrefix());
+      return newMethod.withPrefix(nc.getPrefix());
     }
   }
 }