I read several cases that the person can return to US while being checked as long as visa is not rejected at spot. They stamped something on your I-94 or passport indicting that you are being checked. It would be a nightmare if need to wait in Mexico while being checked.