Add back assert