Fixing the Flee behavior