Fixed minor bug in load script