fix state tests