Fix flying enable/disable bug