Fix remove bug