Remove some TODO: