The mosque has been under extensive restoration since 2009 under the direction of the Aga Khan Trust for Culture and the Government of Punjab, with contributions from the governments of Germany, Norway, and the United States.