diff --git a/test/unit/operations/invoke_host_function_test.js b/test/unit/operations/invoke_host_function_test.js
index c60ebb6c..9c45d5f6 100644
--- a/test/unit/operations/invoke_host_function_test.js
+++ b/test/unit/operations/invoke_host_function_test.js
@@ -63,12 +63,12 @@ describe('Operation', function () {
 
         expect(decodedOp.type).to.equal('invokeHostFunction');
         expect(decodedOp.func.switch().name).to.equal(
-          'hostFunctionTypeCreateContract'
+          'hostFunctionTypeCreateContractV2'
         );
         expect(
           // check deep inner field to ensure RT
           decodedOp.func
-            .createContract()
+            .createContractV2()
             .contractIdPreimage()
             .fromAddress()
             .salt()