diff --git a/main.sh b/main.sh
index 72af9d7..73125ac 100755
--- a/main.sh
+++ b/main.sh
@@ -142,8 +142,8 @@ ebpf() {
 }
 
 containerruntime() {
-	isUbuntu=$(cat /etc/os-release | grep NAME | grep Ubuntu | wc -l)
-	if [ $isUbuntu -gt 0 ]; then
+	isUbuntu=$(grep -c Ubuntu /etc/os-release)
+	if [ "$isUbuntu" -gt 0 ]; then
 		# Add Docker's official GPG key:
 		sudo apt-get update -y
 		sudo apt-get install ca-certificates curl gnupg -y