Novak Djokovic says his body is "feeling good" ahead of his return to the court at this week's China Open, following his first-round exit at the US Open.