fixed removal bug
fiexed removal bug