small update to unit tests