diff --git a/tests/stability/gha-stability-run.sh b/tests/stability/gha-stability-run.sh
index f09d5a5560..16d907e50b 100755
--- a/tests/stability/gha-stability-run.sh
+++ b/tests/stability/gha-stability-run.sh
@@ -12,6 +12,8 @@ set -o pipefail
 stability_dir="$(dirname "$(readlink -f "$0")")"
 source "${stability_dir}/../common.bash"
 source "${stability_dir}/../metrics/lib/common.bash"
+source "${stability_dir}/../gha-run-k8s-common.sh"
+source "${stability_dir}/../integration/kubernetes/gha-run.sh"
 
 function run_tests() {
 	info "Running scability test using ${KATA_HYPERVISOR} hypervisor"
@@ -27,7 +29,18 @@ function run_tests() {
 function main() {
 	action="${1:-}"
 	case "${action}" in
+		install-azure-cli) install_azure_cli ;;
+		login-azure) login_azure ;;
+		create-cluster) create_cluster ;;
+		install-bats) install_bats ;;
+		install-kubectl) install_kubectl ;;
+		get-cluster-credentials) get_cluster_credentials ;;
+		deploy-snapshotter) deploy_snapshotter ;;
+		deploy-kata-aks) deploy_kata "aks" ;;
+		deploy-coco-kbs) deploy_coco_kbs ;;
+		install-kbs-client) install_kbs_client ;;
 		run-tests) run_tests ;;
+		delete-cluster) cleanup "aks" ;;
 		*) >&2 die "Invalid argument" ;;
 	esac
 }