Fix model initialization