model stability fix