Fix afterInstall callback