revert changes to remove method

The changes made to the remove method were unnecessary, so reverting