Fix invalid import