From 18a091f8eaf2dea409b5e528b01ab7abc14ab27a Mon Sep 17 00:00:00 2001
From: Arona Jones <aj@aronajones.com>
Date: Tue, 26 Jan 2021 19:42:54 +0000
Subject: [PATCH] What about this?

---
 assets/tests/test_assets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/assets/tests/test_assets.py b/assets/tests/test_assets.py
index ec4f2f14..fc14bb75 100644
--- a/assets/tests/test_assets.py
+++ b/assets/tests/test_assets.py
@@ -327,7 +327,7 @@ class TestAssetAudit(AutoLoginTest):
         self.assertEqual(self.page.modal.asset_id, asset_row.id)
 
         self.page.modal.close()
-        self.wait.until(EC.invisibility_of_element_located((By.ID, 'modal')))
+        self.wait.until(animation_is_finished())
         self.assertFalse(self.driver.find_element_by_id('modal').is_displayed())
         # Make sure audit log was NOT filled out
         audited = models.Asset.objects.get(asset_id=asset_row.id)