Wrong import statement fix