repairing unittest