corrections to unit tests