Fix findById callback signature