Modified GetStateInfo logic