Fix missing comment