Fix bad conditional import