correct the load function