Fixing wrong import directives