diff --git a/src/org/labkey/test/pages/ReactAssayDesignerPage.java b/src/org/labkey/test/pages/ReactAssayDesignerPage.java
index cb6c641f92..457b7254b3 100644
--- a/src/org/labkey/test/pages/ReactAssayDesignerPage.java
+++ b/src/org/labkey/test/pages/ReactAssayDesignerPage.java
@@ -260,8 +260,8 @@ private ReactAssayDesignerPage setTransformScript(File transformScript, boolean
 
     public ReactAssayDesignerPage removeTransformScript(String fileName)
     {
+        AttachmentCard card = new AttachmentCard.FileAttachmentCardFinder(getDriver()).withFile(fileName).waitFor(this);
         int beforeCount = Locator.tagWithClass("div", "attachment-card__description").findElements(this).size();
-        AttachmentCard card = new AttachmentCard.FileAttachmentCardFinder(getDriver()).withFile(fileName).find(this);
         card.clickRemove();
         int afterCount = Locator.tagWithClass("div", "attachment-card__description").findElements(this).size();
         assertEquals("Transform script count not as expected after remove.", beforeCount - 1, afterCount);