Temporarily fix import issue in LocalTest