diff --git a/containers/Dockerfile b/containers/Dockerfile
index 401f7602..0fd0c85f 100644
--- a/containers/Dockerfile
+++ b/containers/Dockerfile
@@ -14,7 +14,7 @@ COPY --from=azure-cli /usr/local/bin/az /usr/bin/az
 # Install dependencies
 RUN yum install -y git python39 python3-pip jq gettext wget && \
     python3.9 -m pip install -U pip && \
-    git clone https://github.com/redhat-chaos/krkn.git --branch v1.5.0 /root/kraken && \
+    git clone https://github.com/redhat-chaos/krkn.git --branch v1.5.1 /root/kraken && \
     mkdir -p /root/.kube && cd /root/kraken && \
     pip3.9 install -r requirements.txt && \
     pip3.9 install virtualenv && \
diff --git a/containers/Dockerfile-ppc64le b/containers/Dockerfile-ppc64le
index c630ec35..b6eb795d 100644
--- a/containers/Dockerfile-ppc64le
+++ b/containers/Dockerfile-ppc64le
@@ -14,7 +14,7 @@ COPY --from=azure-cli /usr/local/bin/az /usr/bin/az
 # Install dependencies
 RUN yum install -y git python39 python3-pip jq gettext wget && \
     python3.9 -m pip install -U pip && \
-    git clone https://github.com/redhat-chaos/krkn.git --branch v1.5.0 /root/kraken && \
+    git clone https://github.com/redhat-chaos/krkn.git --branch v1.5.1 /root/kraken && \
     mkdir -p /root/.kube && cd /root/kraken && \
     pip3.9 install -r requirements.txt && \
     pip3.9 install virtualenv && \
diff --git a/kraken/node_actions/run.py b/kraken/node_actions/run.py
index 1eff6a14..7b01a7f6 100644
--- a/kraken/node_actions/run.py
+++ b/kraken/node_actions/run.py
@@ -15,6 +15,8 @@
 from krkn_lib.k8s import KrknKubernetes
 from krkn_lib.telemetry.k8s import KrknTelemetryKubernetes
 from krkn_lib.models.telemetry import ScenarioTelemetry
+from krkn_lib.utils.functions import get_yaml_item_value
+
 node_general = False