generator: fix import