Fixed missing flash messages