Fix TODOS