Small fix for remove