diff --git a/plugins/ide-features-trainer/src/training/learn/lesson/general/SurroundAndUnwrapLesson.kt b/plugins/ide-features-trainer/src/training/learn/lesson/general/SurroundAndUnwrapLesson.kt
index 2f88e9116560..f86eca281ecd 100644
--- a/plugins/ide-features-trainer/src/training/learn/lesson/general/SurroundAndUnwrapLesson.kt
+++ b/plugins/ide-features-trainer/src/training/learn/lesson/general/SurroundAndUnwrapLesson.kt
@@ -43,7 +43,7 @@ abstract class SurroundAndUnwrapLesson
             surroundItems.all { sequence.contains(it) }
           }
         }
-        restoreByUi()
+        restoreByUi(delayMillis = defaultRestoreDelay)
         test {
           type("${surroundItems.joinToString(separator = " ")}\n")
         }