Minor correction of unit tests