Fix bad import statement