Vietnam -> Beijing -> Sandiego but a layover in Shanghai.

Hello. My flight is HoChiMin city to Beijing and then out of Beijing to Sandiego 2 days later. However, the flight from Vietnam to Beijing has a stop in Shanghai for a few hours. Will I be able to utilize the transit policy in Beijing like I have planned or does it only apply to Shanghai?
Thank you.