Fixed generated import