Shanghai: Djokovic had to “remove the rust”