Fix comment
Add a test