On 23 July 2017, Khali returned to WWE at Battleground as a heel to assist WWE Champion Jinder Mahal in a Punjabi Prison match for the WWE Championship by preventing Randy Orton from escaping the structure, allowing Mahal to escape and retain the title.