Flash message error fixed