diff --git a/tools/ubuntu_setup.sh b/tools/ubuntu_setup.sh
index 379df37adb..863b853718 100755
--- a/tools/ubuntu_setup.sh
+++ b/tools/ubuntu_setup.sh
@@ -108,7 +108,11 @@ if [ -f "/etc/os-release" ]; then
       if [[ ! $REPLY =~ ^[Yy]$ ]]; then
         exit 1
       fi
-      install_ubuntu_focal_requirements
+      if [ "$UBUNTU_CODENAME" = "jammy" ]; then
+        install_ubuntu_jammy_requirements
+      else
+        install_ubuntu_focal_requirements
+      fi
   esac
 else
   echo "No /etc/os-release in the system"