Fix TODO notice