Added todo

Forgot todo in my last PR