Remove TODO
Remove useless assert