fix: ensure generated code matches expected output

fix: ensure generated code matches expected output #277