fix one bug in load_model handle