Fix bad import