Fix error in the unit tests