Fix generation with latest docket. Fixes #357.