Fixing comment

Forgot to update comment after function change