Fixed forgotten comments