diff --git a/pkg/wasm/main_test.go b/pkg/wasm/main_test.go
index c02d254..403d1a3 100644
--- a/pkg/wasm/main_test.go
+++ b/pkg/wasm/main_test.go
@@ -56,8 +56,8 @@ func TestZedCommand(t *testing.T) {
 	require.NoError(t, err)
 
 	require.Contains(t, updatedContext.Schema, "definition document")
-	require.Equal(t, `document:first#viewer@user:fred[somecaveat:{"somecondition":42}]`, tuple.CoreRelationToString(updatedContext.Relationships[0]))
-	require.Equal(t, "document:first#viewer@user:tom", tuple.CoreRelationToString(updatedContext.Relationships[1]))
+	require.Equal(t, `document:first#viewer@user:fred[somecaveat:{"somecondition":42}]`, stringifyTuple(updatedContext.Relationships[0]))
+	require.Equal(t, "document:first#viewer@user:tom", stringifyTuple(updatedContext.Relationships[1]))
 	require.Len(t, updatedContext.Relationships, 2)
 
 	// Run the actual command.
@@ -69,3 +69,9 @@ func TestZedCommand(t *testing.T) {
 	require.NoError(t, err)
 	require.Len(t, updatedContext.Relationships, 3)
 }
+
+// NOTE: this is required to preserve caveat context.
+// tuple.CoreRelationToString elides the caveat context.
+func stringifyTuple(relationtuple *v1.RelationTuple) string {
+	return tuple.MustString(tuple.FromCoreRelationTuple(relationtuple))
+}