bugfix: update onRemove to correspond to recent onAdd changes

I changed onRemove to correspond to your recent zIndex fixes.