First, I think INS made mistake in your wife's case.Second, it is better for your wife to go back to China right away. And apply visa as your spouse to re-enter USA.It is a BAD idea to wait until the baby comes out while she is out of status.