Flash messages fixed