Fix error in codegen
TBR=@jmesserly