Add ensure parameter