diff --git a/src/Blt/Plugin/Commands/DrushAliasCommand.php b/src/Blt/Plugin/Commands/DrushAliasCommand.php
index 0d319d6..ea64fbe 100644
--- a/src/Blt/Plugin/Commands/DrushAliasCommand.php
+++ b/src/Blt/Plugin/Commands/DrushAliasCommand.php
@@ -5,8 +5,8 @@
 use Acquia\Blt\Robo\BltTasks;
 use Acquia\Blt\Robo\Exceptions\BltException;
 use Acquia\Hmac\Exception\MalformedResponseException as MalformedResponseExceptionAlias;
-use AcquiaCloudApi\CloudApi\Client;
-use AcquiaCloudApi\CloudApi\Connector;
+use AcquiaCloudApi\Connector\Client;
+use AcquiaCloudApi\Connector\Connector;
 use AcquiaCloudApi\Endpoints\Account;
 use AcquiaCloudApi\Endpoints\Applications;
 use AcquiaCloudApi\Endpoints\Environments;
@@ -21,7 +21,7 @@ class DrushAliasCommand extends BltTasks {
   /**
    * Cloud API client.
    *
-   * @var \AcquiaCloudApi\CloudApi\Client
+   * @var \AcquiaCloudApi\Connector\Client
    */
   protected $cloudApiClient;
   /**
@@ -196,7 +196,7 @@ protected function setCloudApiClient($key, $secret) {
       ]);
       $cloud_api = Client::factory($connector);
       // We must call some method on the client to test authentication.
-      $account = new Account();
+      $account = new Account($cloud_api);
       $account->get();
       $this->cloudApiClient = $cloud_api;
     }
@@ -215,7 +215,7 @@ protected function setCloudApiClient($key, $secret) {
   /**
    * Gets connection with API client.
    *
-   * @return \AcquiaCloudApi\CloudApi\Client
+   * @return \AcquiaCloudApi\Connector\Client
    *   The API Client connection.
    */
   protected function getCloudApiClient() {