fix model reference