Restore OKButton reference