I'm not sure this is really needed. We could move all this to the `load` method maybe.