Fixing bug in model load