Small fix for comments