FIX test: hack around the inference... maybe we don't want to revert back to the original inference after all?