changes to unit tests