Fix remove method doesn't exist