fixed tests to reflect change of remove button functionality