Fixed a missing comment