Modify the load method