Fix load bug