Fix model reg