1 files changed, 8 insertions, 8 deletions
diff --git a/node_modules/selenium-webdriver/test/chrome/options_test.js b/node_modules/selenium-webdriver/test/chrome/options_test.js
index 28c4faa91..d36a044e4 100644
--- a/node_modules/selenium-webdriver/test/chrome/options_test.js
+++ b/node_modules/selenium-webdriver/test/chrome/options_test.js
@@ -205,22 +205,22 @@ test.suite(function(env) {
   var driver;
 
   test.afterEach(function() {
-    driver.quit();
+    return driver.quit();
   });
 
   describe('Chrome options', function() {
-    test.it('can start Chrome with custom args', function() {
+    test.it('can start Chrome with custom args', function*() {
       var options = new chrome.Options().
           addArguments('user-agent=foo;bar');
 
-      driver = env.builder().
-          setChromeOptions(options).
-          build();
+      driver = yield env.builder()
+          .setChromeOptions(options)
+          .build();
 
-      driver.get(test.Pages.ajaxyPage);
+      yield driver.get(test.Pages.ajaxyPage);
 
-      var userAgent = driver.executeScript(
-          'return window.navigator.userAgent');
+      var userAgent =
+          yield driver.executeScript('return window.navigator.userAgent');
       assert(userAgent).equalTo('foo;bar');
     });
   });