From 5861d0f9b20e227e9c22c520de4e1f480d5092c7 Mon Sep 17 00:00:00 2001
From: Alexey Masterov <alexey.masterov@databricks.com>
Date: Thu, 17 Jul 2025 12:01:50 +0200
Subject: [PATCH] Add the environment

---
 test_runner/random_ops/test_random_ops.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test_runner/random_ops/test_random_ops.py b/test_runner/random_ops/test_random_ops.py
index 13df642d1e..35a657ec56 100644
--- a/test_runner/random_ops/test_random_ops.py
+++ b/test_runner/random_ops/test_random_ops.py
@@ -654,7 +654,8 @@ def test_api_random(
             "psql",
             "-c",
             "CREATE TABLE IF NOT EXISTS sanity_check (name VARCHAR NOT NULL PRIMARY KEY, value VARCHAR)",
-        ]
+        ],
+        env=project.main_branch.connect_env,
     )
     # To not go to the past where pgbench tables do not exist
     time.sleep(1)