Fixed bug stopping remove from working