Fixed bug where load didn't return self