Remove TODO