Fix init_state validation