I want to do it some day for this very reason. I don't think it's possible to fly back the same day, so you'll either need to spend a night (presumably on the Chinese side) or grab a taxi back to Beijing. There are a few good trip reports floating around the web if you search for Erlian and visa.