bug fix on get_states()