Forgot to update TODO