moved a line to get Model loading