Add extra assert