Perhaps obvious in hindsight but in that situation I definitely would not leave the country and especially try to re-enter the country unless my paperwork was

Perhaps obvious in hindsight but in that situation I definitely would not leave the country and especially try to re-enter the country unless my paperwork was in perfect order (lawyer prepared it for you). One mistake like that can ruin your visa situation forever.