Small fix with comments