Fix comment tests