fixed a bug with the remove button