Should fix error with flash message popping up again