Fixes comment to reflect code update