This should be a bit more clear and less prone to potential breakage if the logic of the load models changes a bit.