Fix build break in code generator.

Reviewed at https://reviews.apache.org/r/23087/