fix little state bug