fix model output