Add assertions for output stage cr0