Whoops, missed one in the TODO cleanup