Fix assertion at the end of a generation.

Fixup broken references to ARENA_SCALE.