minor edits to tests