Fix task rendering

Update flash messages on task done.