Minor corrections to 2 unit tests