fix import in maintenance tests