Add a specific step to check for flash messages