Add ensure statement