diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/TestCaseGenerator.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/TestCaseGenerator.kt
index d16101086c..823dae8c8b 100644
--- a/utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/TestCaseGenerator.kt
+++ b/utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/TestCaseGenerator.kt
@@ -184,8 +184,11 @@ open class TestCaseGenerator(
                                         is UtError -> method2errors.getValue(method).merge(it.description, 1, Int::plus)
                                     }
                                 }
+                        } catch (e: CancellationException) {
+                            logger.info(e) { "Test generation cancelled" }
+                            throw(e)
                         } catch (e: Exception) {
-                            logger.error(e) {"Error in engine"}
+                            logger.error(e) { "Error in engine" }
                         }
                     }
                     controller.paused = true