More TODO fixes