Fix remove todo