Add back a comment which went missing during refactoring