Found way to make remove work

Thanks to help from https://github.com/matelich