Updating the API to the new API

JAVA-1325
I found out I can use {{ expect operation, isTheSameAs(expectedOperation)}} so have updated