Fix flood bug