Add missing param, fix variable reference