diff --git a/utils.go b/utils.go
index 5a53cc1..a7d3f67 100644
--- a/utils.go
+++ b/utils.go
@@ -102,13 +102,14 @@ func GetLSM() (lsm string, err error) {
 }
 
 func getPlatformVersion(output, platform string) string {
-	flagRe := regexp.MustCompile("version_id" + `=([^ \n]*)`)
+	flagRe := regexp.MustCompile(`version[_id]*=([^ \n]*)`)
 	vals := flagRe.FindStringSubmatch(output)
-
 	if len(vals) > 1 {
 		switch platform {
 		case "rhel":
 			return vals[1][:1] // Get the major version only, examaple: 7.6 will return 7
+		case "ubuntu":
+			return vals[1][:2] // Get the major version only, examaple: 18.04 will return 18
 		default:
 			return ""
 		}