Fixing bug. Problem in the instance of removal.