Fix docs for test utils
Update docs