Fixed up code generation.