repairing 2 unit tests