Add TODO

Should fix this ASAP, but still can't find what is the problem