Fix TODO header