fixing broken code generation caused by last commit