144 hr free transit for a US citizen traveling in Japan (tourist).

I would like to visit Shanghai for 3 days, and then return to Japan. I understand I qualify for 144 hour free transit exemption, but is it ok to return to Japan, or do I need to return to a "third country?"

Thanks!