fix comment, add tests