Update loadDesign.js

Author: aabidsofi19

cypress-action/cypress/e2e/e2e/loadDesign.js

@@ -56,7 +56,7 @@ const captureSnapshot = ({ window, designId, theme }) => {
     const path = snapshotPath(designId, theme);
     cy.wait(2000);
 
-    cy.get(canvasContainer.query, { timeout: 10 * 1000 })
+    cy.get("main", { timeout: 10 * 1000 })
         .should("exist")
         .screenshot(path, {
             //blackout: [".hide-from-snapshot"], // hides elements before screenshot