Clean model loading a bit.