fix model stability