captureEntirePage doesn't work for long pages

Hi there,
I am new to UI.vision and I’m trying to create a simple script that can capture a screenshot of a large ecommerce web site. Whenever I use a search that has a few results the captureEntirePage works fine, but when there are many results I get the following error:
[warning] Web page connection issue. Retrying last command.
Below is the JSON of something that doesn’t work, if anyone has any tips of tricks to make this work better it would be appreciated.

{
  "Name": "walmartScreenshot",
  "CreationDate": "2019-10-30",
  "Commands": [
    {
      "Command": "open",
      "Target": "https://www.walmart.com/",
      "Value": ""
    },
    {
      "Command": "type",
      "Target": "id=global-search-input",
      "Value": "shirts"
    },
    {
      "Command": "click",
      "Target": "xpath=//*[@id=\"global-search-submit\"]/span/span",
      "Value": ""
    },
    {
      "Command": "waitForPageToLoad",
      "Target": "",
      "Value": ""
    },
    {
      "Command": "captureEntirePageScreenshot",
      "Target": "walmart-search",
      "Value": ""
    },
    {
      "Command": "localStorageExport",
      "Target": "walmart-search.png",
      "Value": ""
    }
  ]
}

Issue confirmed, we are working on it. Thanks for the good test macro!