Fix remove button