Custom template not working after recent upgrade