diff --git a/lib/appium.js b/lib/appium.js
index 5ffb054e4..c1e9890ef 100644
--- a/lib/appium.js
+++ b/lib/appium.js
@@ -7,7 +7,7 @@ import { FakeDriver } from 'appium-fake-driver';
 import { AndroidDriver } from 'appium-android-driver';
 import { IosDriver } from 'appium-ios-driver';
 import { SelendroidDriver } from 'appium-selendroid-driver';
-import { WebDriverAgentDriver } from 'appium-xcuitest-driver';
+import { XCUITestDriver } from 'appium-xcuitest-driver';
 import { YouiEngineDriver } from 'appium-youiengine-driver';
 import B from 'bluebird';
 import util from 'util';
@@ -48,7 +48,7 @@ class AppiumDriver extends BaseDriver {
         return SelendroidDriver;
       } else if (caps.automationName.toLowerCase() === 'xcuitest') {
         // but if we do and it is 'XCUITest', act on it
-        return WebDriverAgentDriver;
+        return XCUITestDriver;
       } else if (caps.automationName.toLowerCase() === 'youiengine') {
         // but if we do and it is 'YouiEngine', act on it
         return YouiEngineDriver;
diff --git a/package.json b/package.json
index fcf0f5d3e..95f2202ae 100644
--- a/package.json
+++ b/package.json
@@ -42,7 +42,7 @@
     "appium-logger": "^2.1.0",
     "appium-selendroid-driver": "^1.3.4",
     "appium-support": "^2.3.0",
-    "appium-xcuitest-driver": "^1.0.4",
+    "appium-xcuitest-driver": "^2.0.0",
     "appium-youiengine-driver": "^1.0.3",
     "argparse": "^1.0.7",
     "asyncbox": "^2.3.1",
diff --git a/test/driver-specs.js b/test/driver-specs.js
index ae58d7eb2..ec00877a0 100644
--- a/test/driver-specs.js
+++ b/test/driver-specs.js
@@ -7,7 +7,7 @@ import _ from 'lodash';
 import sinon from 'sinon';
 import chai from 'chai';
 import chaiAsPromised from 'chai-as-promised';
-import { WebDriverAgentDriver } from 'appium-xcuitest-driver';
+import { XCUITestDriver } from 'appium-xcuitest-driver';
 
 
 chai.use(chaiAsPromised);
@@ -176,14 +176,14 @@ describe('AppiumDriver', () => {
         let appium = new AppiumDriver({});
         (() => { appium.getDriverForCaps({}); }).should.throw(/platformName/);
       });
-      it('should get WebDriverAgentDriver driver for automationName of XCUITest', () => {
+      it('should get XCUITestDriver driver for automationName of XCUITest', () => {
         let appium = new AppiumDriver({});
         let driver = appium.getDriverForCaps({
           platformName: 'iOS',
           automationName: 'XCUITest'
         });
         driver.should.be.an.instanceof(Function);
-        driver.should.equal(WebDriverAgentDriver);
+        driver.should.equal(XCUITestDriver);
       });
     });
   });