diff --git a/up/tests/up_tests.py b/up/tests/up_tests.py
index a04fddc..40f1ed0 100644
--- a/up/tests/up_tests.py
+++ b/up/tests/up_tests.py
@@ -171,7 +171,7 @@ def test_start_azure_openai_smoketest():
     kernel_key = random_str(32)
     with Live(Group(*segments), console=console) as live:
         generate_kernel_env(live, segments, install_dir, kernel_key)
-        code = load_docker_image("v0.5.1", "dbpunk/octogen", live, segments)
+        code = load_docker_image("v0.5.2", "dbpunk/octogen", live, segments)
         assert code == 0, "bad result code of loading docker image"
         result = start_octogen_for_azure_openai(
             live,
@@ -181,7 +181,7 @@ def test_start_azure_openai_smoketest():
             admin_key,
             kernel_key,
             "dbpunk/octogen",
-            "v0.5.1",
+            "v0.5.2",
             "azure_open_api_key",
             "test_deployment",
             "https://azure_base",
@@ -200,7 +200,7 @@ def test_start_openai_smoketest():
     kernel_key = random_str(32)
     with Live(Group(*segments), console=console) as live:
         generate_kernel_env(live, segments, install_dir, kernel_key)
-        code = load_docker_image("v0.5.1", "dbpunk/octogen", live, segments)
+        code = load_docker_image("v0.5.2", "dbpunk/octogen", live, segments)
         assert code == 0, "bad result code of loading docker image"
         result = start_octogen_for_openai(
             live,
@@ -210,7 +210,7 @@ def test_start_openai_smoketest():
             admin_key,
             kernel_key,
             "dbpunk/octogen",
-            "v0.5.1",
+            "v0.5.2",
             "openai_api_key",
             "gpt-3.5-turbo",
         )
@@ -227,7 +227,7 @@ def test_start_codellama_smoketest():
     kernel_key = random_str(32)
     with Live(Group(*segments), console=console) as live:
         generate_kernel_env(live, segments, install_dir, kernel_key)
-        code = load_docker_image("v0.5.1", "dbpunk/octogen", live, segments)
+        code = load_docker_image("v0.5.2", "dbpunk/octogen", live, segments)
         assert code == 0, "bad result code of loading docker image"
         result = start_octogen_for_codellama(
             live,
@@ -239,7 +239,7 @@ def test_start_codellama_smoketest():
             admin_key,
             kernel_key,
             "dbpunk/octogen",
-            "v0.5.1",
+            "v0.5.2",
         )
         assert result
 
@@ -258,5 +258,5 @@ def test_load_valid_docker_image():
     console = Console()
     segments = []
     with Live(Group(*segments), console=console) as live:
-        code = load_docker_image("v0.5.1", "dbpunk/octogen", live, segments)
+        code = load_docker_image("v0.5.2", "dbpunk/octogen", live, segments)
         assert code == 0, "loading image should be ok"