Similar way to what was done for `SingleOmegaModel`
Similar way to what was done for
SingleOmegaModel