small changes to unit tests