Fixing context of the load function