Fix bad imports