Updated comment
Forgot to update this in fix for #538