Fix remove button styling