From c631dffd1c4f4d4d4cd0ba0f1dbc55a1adb902e7 Mon Sep 17 00:00:00 2001
From: Robin Tang <rtang.cs@gmail.com>
Date: Fri, 20 Dec 2024 14:43:42 -0800
Subject: [PATCH] Fix integration tests

---
 integration_tests/mongo/main.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/integration_tests/mongo/main.go b/integration_tests/mongo/main.go
index 1e3650240..f49897c22 100644
--- a/integration_tests/mongo/main.go
+++ b/integration_tests/mongo/main.go
@@ -122,7 +122,7 @@ func testTypes(ctx context.Context, db *mongo.Database, mongoCfg config.MongoDB)
 	}
 
 	row := rows[0]
-	expectedPartitionKey := map[string]any{"payload": map[string]any{"id": `{"$oid":"66a95fae3776c2f21f0ff568"}`}}
+	expectedPartitionKey := map[string]any{"id": `{"$oid":"66a95fae3776c2f21f0ff568"}`}
 	expectedPkBytes, err := json.Marshal(expectedPartitionKey)
 	if err != nil {
 		return fmt.Errorf("failed to marshal expected partition key: %w", err)