By late 20th century, the magnificent tomb had fallen into a state of dilapidation. The masonry and stonework were broken and the gardens were run down. In 1997, research for restoration work was taken up by the Aga Khan Trust for Culture (AKTC), in collaboration with the Archaeological Survey of India (ASI). Restoration began in 1999 and was completed in 2004, with the gardens fully restored as well.