diff --git a/gen3-integration-tests/pages/login.py b/gen3-integration-tests/pages/login.py
index 70f7d322..01031a2f 100644
--- a/gen3-integration-tests/pages/login.py
+++ b/gen3-integration-tests/pages/login.py
@@ -104,7 +104,7 @@ def login(
             ):
                 page.locator(self.USER_PROFILE_DROPDOWN).click()
             expect(
-                page.locator(f'//div[contains(text(), "{logged_in_user}")]')
+                page.get_by_role("link").filter(has_text=logged_in_user)
             ).to_be_visible(timeout=10000)
         screenshot(page, "AfterLogin")
         self.handle_popup(page)
diff --git a/gen3-integration-tests/services/audit.py b/gen3-integration-tests/services/audit.py
index 20f88afc..e991d53d 100644
--- a/gen3-integration-tests/services/audit.py
+++ b/gen3-integration-tests/services/audit.py
@@ -1,6 +1,5 @@
 import datetime
 import math
-import os
 import pytest
 import time