diff --git a/supervisely/train/src/sly_train.py b/supervisely/train/src/sly_train.py
index 7597d4b2a4d7..5cda62a4b42b 100644
--- a/supervisely/train/src/sly_train.py
+++ b/supervisely/train/src/sly_train.py
@@ -128,11 +128,11 @@ def train(api: sly.Api, task_id, context, state, app_logger):
         upload_artifacts(g.local_artifacts_dir, g.remote_artifacts_dir)
         set_task_output()
     except Exception as e:
-        # msg = f"Something went wrong. Find more info in the app logs. {repr(e)}"
-        # my_app.show_modal_window(msg, level="error")
-        # sly.logger.error(msg, exc_info=True, extra={ 'exc_str': str(e)})
-        # api.app.set_field(task_id, "state.started", False)
-        raise e
+        msg = f"Something went wrong. Find more info in the app logs. {str(e)}"
+        my_app.show_modal_window(msg, level="error", log_message=False)
+        sly.logger.error(msg, exc_info=True, extra={ 'exc_str': str(e)})
+        api.task.set_output_error(task_id, msg)
+        api.app.set_field(task_id, "state.started", False)
 
     # stop application
     get_progress_cb("Finished, app is stopped automatically", 1)(1)