Fix model register