diff --git a/imagetest/test_suites/network/gvnic_test.go b/imagetest/test_suites/network/gvnic_test.go
index 1453c4e58..2896c261c 100644
--- a/imagetest/test_suites/network/gvnic_test.go
+++ b/imagetest/test_suites/network/gvnic_test.go
@@ -26,8 +26,8 @@ func CheckGVNICPresent(interfaceName string) error {
 	}
 	s := strings.Split(data, "/")
 	driver := s[len(s)-1]
-	if driver != "gvnic" {
-		errMsg := fmt.Sprintf("Driver set as %v want gvnic", driver)
+	if driver != "gvnic" && driver != "gve" {
+		errMsg := fmt.Sprintf("Driver set as %v want gvnic or gve", driver)
 		return errors.New(errMsg)
 	}
 	return nil
diff --git a/imagetest/test_suites/networkperf/gvnic_test.go b/imagetest/test_suites/networkperf/gvnic_test.go
index 17ba06c0d..141f947c7 100644
--- a/imagetest/test_suites/networkperf/gvnic_test.go
+++ b/imagetest/test_suites/networkperf/gvnic_test.go
@@ -26,8 +26,8 @@ func CheckGVNICPresent(interfaceName string) error {
 	}
 	s := strings.Split(data, "/")
 	driver := s[len(s)-1]
-	if driver != "gvnic" {
-		errMsg := fmt.Sprintf("Driver set as %v want gvnic", driver)
+	if driver != "gvnic" && driver != "gve" {
+		errMsg := fmt.Sprintf("Driver set as %v want gvnic or gve", driver)
 		return errors.New(errMsg)
 	}
 	return nil