Fix missing comments