From c97a0ce30a3ee946ba01e4d3c1d4a782b28af257 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Thu, 1 Jun 2023 08:48:55 -0700
Subject: [PATCH] wip

---
 tests/test_main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_main.py b/tests/test_main.py
index e79d9ef39..a972be008 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -21,8 +21,8 @@ class TestMain(TestCase):
         self.tempdir = tempfile.mkdtemp()
         os.chdir(self.tempdir)
 
-    with patch("aider.main.Coder.check_model_availability") as mock_check:
-        mock_check.return_value = True
+        self.mock_check = patch("aider.main.Coder.check_model_availability")
+        self.mock_check.return_value = True
 
     def tearDown(self):
         os.chdir(self.original_cwd)