fix code broken by previous commit