Move TODO comment