fix up the code generation tests