Bugfix ensure op

Fixup tests