diff --git a/scripts/upgrade_kernel.sh b/scripts/upgrade_kernel.sh
index 232d59c22..5736d0f5c 100755
--- a/scripts/upgrade_kernel.sh
+++ b/scripts/upgrade_kernel.sh
@@ -13,12 +13,7 @@ fi
 if [[ $KERNEL_VERSION == "4.14" ]]; then
   sudo yum update -y kernel
 elif [[ $KERNEL_VERSION == "5.4" ]]; then
-  # Pinning Kernel to 5.4.209-116.367 since we're investigating issues with later Kernel versions which cause nodes to become Unready.
-  # sudo amazon-linux-extras install -y kernel-5.4
-  sudo amazon-linux-extras enable kernel-5.4=latest
-  sudo yum -y install kernel-5.4.209-116.367.amzn2
-  sudo yum install -y yum-plugin-versionlock
-  sudo yum versionlock kernel-5.4*
+  sudo amazon-linux-extras install -y kernel-5.4
 elif [[ $KERNEL_VERSION == "5.10" ]]; then
   sudo amazon-linux-extras install -y kernel-5.10
 else