repaired import error in test