Had to make a change to the unit tests to get them to compile