Generator fixes.

Fix #63.