Fixed a missing comment block related to #336