unit test minor edits