fix callback for remove method