Fix broken code generation response