fix model wrapping