[minor] Remove TODO