Adapt to the new callback function scheme in mirage-types