From f505ece25fa449e5fb323e77c858372ffb2084c4 Mon Sep 17 00:00:00 2001
From: Noah McGregor Harper <74685766+nharper285@users.noreply.github.com>
Date: Fri, 7 Jan 2022 13:10:10 -0800
Subject: [PATCH] Fixing proxy tag issue. (#1568)

---
 src/api-service/__app__/onefuzzlib/azure/vm.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/api-service/__app__/onefuzzlib/azure/vm.py b/src/api-service/__app__/onefuzzlib/azure/vm.py
index e925785ec..4f89311e9 100644
--- a/src/api-service/__app__/onefuzzlib/azure/vm.py
+++ b/src/api-service/__app__/onefuzzlib/azure/vm.py
@@ -116,12 +116,11 @@ def create_vm(
             },
         }
 
-    if tags:
-        params["tags"] = tags.copy()
+    if "ONEFUZZ_OWNER" in os.environ:
+        params["tags"] = {"OWNER": os.environ["ONEFUZZ_OWNER"]}
 
-    owner = os.environ.get("ONEFUZZ_OWNER")
-    if owner:
-        params["tags"]["OWNER"] = owner
+    if tags:
+        params["tags"].update(tags.copy())
 
     try:
         compute_client.virtual_machines.begin_create_or_update(