fixing generated import