Fixing remove buttom