diff --git a/examples/vmwithcontracts/actions/transfer_test.go b/examples/vmwithcontracts/actions/transfer_test.go
index dfdecbf54b..2b016fa3a1 100644
--- a/examples/vmwithcontracts/actions/transfer_test.go
+++ b/examples/vmwithcontracts/actions/transfer_test.go
@@ -52,10 +52,16 @@ func TestTransferAction(t *testing.T) {
 				Value: 1,
 			},
 			State: func() state.Mutable {
-				keys := make(state.Keys)
-				keys.Add(string(emptyBalanceKey), state.Read)
-				tsv := ts.NewView(keys, map[string][]byte{})
-				return tsv
+				s := chaintest.NewInMemoryStore()
+				_, err := storage.AddBalance(
+					context.Background(),
+					s,
+					codec.EmptyAddress,
+					0,
+					true,
+				)
+				require.NoError(t, err)
+				return s
 			}(),
 			ExpectedErr: storage.ErrInvalidBalance,
 		},
@@ -79,6 +85,10 @@ func TestTransferAction(t *testing.T) {
 				require.NoError(err)
 				require.Equal(balance, uint64(1))
 			},
+			ExpectedOutputs: &TransferResult{
+				SenderBalance:   0,
+				ReceiverBalance: 1,
+			},
 		},
 		{
 			Name:  "OverflowBalance",
@@ -120,6 +130,10 @@ func TestTransferAction(t *testing.T) {
 				require.NoError(err)
 				require.Equal(senderBalance, uint64(0))
 			},
+			ExpectedOutputs: &TransferResult{
+				SenderBalance:   0,
+				ReceiverBalance: 1,
+			},
 		},
 	}