Novak Djokovic's wrist injury is not as serious as first feared and he could play at the Madrid Masters in two weeks' time.