From b9a0c5080a50e177726c2fdf6eb7cf377d25bfc9 Mon Sep 17 00:00:00 2001
From: Jiaming Xu <jiamingxu@google.com>
Date: Mon, 4 Oct 2021 16:36:44 +0000
Subject: [PATCH] modify the way for checking cos

Signed-off-by: Jiaming Xu <jiamingxu@google.com>

address feedback

remove empty spaces

address feedbacks
---
 test/e2e_node/gci-init.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/e2e_node/gci-init.sh b/test/e2e_node/gci-init.sh
index 311cfd042..c24be53b7 100755
--- a/test/e2e_node/gci-init.sh
+++ b/test/e2e_node/gci-init.sh
@@ -26,7 +26,10 @@ fi
 
 if [ "${CONTAINERD_CGROUPV2:-"false"}"  == "true" ]; then
   # check cos image
-  if uname -a | grep -q cos; then
+  if [ -r /etc/os-release ]; then
+    OS_ID="$(. /etc/os-release && echo "$ID")"
+  fi
+  if [ "${OS_ID}" = "cos" ]; then
     if ! grep -q 'systemd.unified_cgroup_hierarchy=true' /proc/cmdline; then
       echo "Setting up cgroupv2"