diff --git a/roles/custom/matrix-bot-baibot/defaults/main.yml b/roles/custom/matrix-bot-baibot/defaults/main.yml
index ab1770abd..83661afca 100644
--- a/roles/custom/matrix-bot-baibot/defaults/main.yml
+++ b/roles/custom/matrix-bot-baibot/defaults/main.yml
@@ -13,7 +13,7 @@ matrix_bot_baibot_enabled: true
 
 matrix_bot_baibot_container_image_self_build: false
 matrix_bot_baibot_container_repo: "https://github.com/etkecc/baibot.git"
-matrix_bot_baibot_container_repo_version: "{{ 'main' if matrix_bot_baibot_version == 'latest' else matrix_bot_baibot_version }}"
+matrix_bot_baibot_container_repo_version: "{{ 'main' if matrix_bot_baibot_version in ['latest', 'edge'] else matrix_bot_baibot_version }}"
 matrix_bot_baibot_container_src_files_path: "{{ matrix_base_data_path }}/baibot/container-src"
 
 # renovate: datasource=docker depName=ghcr.io/etkecc/baibot