Update callback signature