Fix comment
Add test