Fix callback signature