Codegen: clean-up imports