fix import error when launching tests