Bug fix for load model.