fixed two broken unit tests