Rewrite SessionModelWrapper functions