Fixes import path in generated code