Fixes done message not shown once at end.