From 672111635fd587e05a708b05f85a1ee78abb56db Mon Sep 17 00:00:00 2001
From: yann300 <yann.levreau@gmail.com>
Date: Mon, 11 Dec 2017 14:46:13 +0100
Subject: [PATCH] fix test: addInstance function

---
 test-browser/helpers/contracts.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test-browser/helpers/contracts.js b/test-browser/helpers/contracts.js
index 2e8ef28c40..b0e6328c04 100644
--- a/test-browser/helpers/contracts.js
+++ b/test-browser/helpers/contracts.js
@@ -147,15 +147,17 @@ function setEditorValue (value) {
   return this
 }
 
-function addInstance (browser, address, done) {
+function addInstance (browser, address, callback) {
   browser.setValue('.ataddressinput', address, function () {
     browser.click('div[class^="atAddress"]')
-      .perform((client) => {
+      .perform((client, done) => {
         browser.execute(function () {
           document.querySelector('#modal-footer-ok').click()
         }, [], function (result) {
           done()
         })
+      }).perform(() => {
+        callback()
       })
   })
 }