Update to when Yii is loaded