Fixed bug in load function.