diff --git a/upgrade_edge/upgrade_edge.sh b/upgrade_edge/upgrade_edge.sh
index eb3031d..2a111fd 100755
--- a/upgrade_edge/upgrade_edge.sh
+++ b/upgrade_edge/upgrade_edge.sh
@@ -8,6 +8,7 @@
      # UPGRADE_KUBERENTES="true"
 
  echo $UPGRADE_KUBERNETES
+ echo $TALOS_VERSION
 if [ "$UPGRADE_KUBERNETES" != "true" ] && [ "$UPGRADE_TALOS" != "true" ]; then
   echo "None of the variables UPGRADE_KUBERNETES or UPGRADE_TALOS has been set to true (true as a string, not a boolean), exiting without any upgrading"
   echo $UPGRADE_KUBERENTES
@@ -85,6 +86,7 @@ elif [ "$UPGRADE_TALOS" = "true" ]; then
     check_talosctl_version
     upgrade_talos
 elif [ "$UPGRADE_KUBERNETES" = "true" ]; then
+    check_talos_version_existence
     check_talosctl_version
     get_current_k8s_version
     upgrade_kubernetes