Edits to unit tests.