Fix module generator.

fixes #672