Fixes minor logic issue with list removal