fix import error introduced during minor cleanup