repaired two unit tests