Fix loadMOdel