BELGRADE, Serbia (AP) — Novak Djokovic's team says Andre Agassi will remain his coach in 2018.

The 12-time Grand Slam champion stepped off the ATP Tour following Wimbledon, where he retired during his quarterfinal against Tomas Berdych because of an elbow injury.

Djokovic brought Agassi into his team in May before the French Open. They worked together for two months. After Wimbledon, there were doubts the American great would continue coaching the Serb.

Djokovic hired Agassi to repair his form, and lost in the quarterfinals of the French Open and Wimbledon, and won in between at Eastbourne for his first title in six months.

He is expected to return to play at the Australian Open in January.