Now that I've moved to marionette.Wait I have changed this.