Fixes remove button