Adapted for the latest model initialization changes.