diff --git a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.fir.kt b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.fir.kt
index 769fbcd6523..4c42b3270cb 100644
--- a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.fir.kt
+++ b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.fir.kt
@@ -5,9 +5,12 @@ interface I {
 }
 
 // FILE: JavaClass.java
+import kotlin.coroutines.Continuation;
+import org.jetbrains.annotations.NotNull;
+
 public class JavaClass implements I {
     @Override
-    public String foo(int x, kotlin.coroutines.Continuation<String> continuation) {
+    public Object foo(int x, @NotNull Continuation<? super String> continuation) {
         return null;
     }
 }
@@ -22,7 +25,7 @@ class K2 : JavaClass() {
 }
 
 class K3 : JavaClass() {
-    override fun foo(x: Int, y: Continuation<String>): <!RETURN_TYPE_MISMATCH_ON_OVERRIDE!>Any?<!> = null
+    override fun foo(x: Int, y: Continuation<String>): Any? = null
 }
 
 fun builder(block: suspend () -> Unit) {}
diff --git a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.kt b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.kt
index 25be3a644e5..57f098269d1 100644
--- a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.kt
+++ b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendCovarianJavaOverride.kt
@@ -5,9 +5,12 @@ interface I {
 }
 
 // FILE: JavaClass.java
+import kotlin.coroutines.Continuation;
+import org.jetbrains.annotations.NotNull;
+
 public class JavaClass implements I {
     @Override
-    public String foo(int x, kotlin.coroutines.Continuation<String> continuation) {
+    public Object foo(int x, @NotNull Continuation<? super String> continuation) {
         return null;
     }
 }
diff --git a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.fir.kt b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.fir.kt
index 2ffcd3ef844..4c42b3270cb 100644
--- a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.fir.kt
+++ b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.fir.kt
@@ -5,9 +5,12 @@ interface I {
 }
 
 // FILE: JavaClass.java
+import kotlin.coroutines.Continuation;
+import org.jetbrains.annotations.NotNull;
+
 public class JavaClass implements I {
     @Override
-    public Object foo(int x, kotlin.coroutines.Continuation<String> continuation) {
+    public Object foo(int x, @NotNull Continuation<? super String> continuation) {
         return null;
     }
 }
diff --git a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.kt b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.kt
index ba4ed507369..57f098269d1 100644
--- a/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.kt
+++ b/compiler/testData/diagnostics/testsWithStdLib/coroutines/suspendJavaOverrides.kt
@@ -5,9 +5,12 @@ interface I {
 }
 
 // FILE: JavaClass.java
+import kotlin.coroutines.Continuation;
+import org.jetbrains.annotations.NotNull;
+
 public class JavaClass implements I {
     @Override
-    public Object foo(int x, kotlin.coroutines.Continuation<String> continuation) {
+    public Object foo(int x, @NotNull Continuation<? super String> continuation) {
         return null;
     }
 }