diff --git a/tests/unit/Client/TestClientFactory.php b/tests/unit/Client/TestClientFactory.php
index 83dc5a9c6ed6a4417111e4b2b3f8d6b55c6277e5..553524d74164ba77435805ca91dbc51f0cbd4dcc 100644
--- a/tests/unit/Client/TestClientFactory.php
+++ b/tests/unit/Client/TestClientFactory.php
@@ -15,6 +15,10 @@ class TestClientFactory extends \PHPUnit\Framework\TestCase
     private function prepareOptions($isCachedClient = false)
     {
         $options = Mockery::mock(\WPDesk\ApiClient\Client\ApiClientOptions::class);
+        $options->shouldReceive('getApiClientClass')
+                ->withAnyArgs()
+                ->andReturn(\WPDesk\ApiClient\Client\ClientImplementation::class);
+
         $options->shouldReceive('getHttpClientClass')
                 ->withAnyArgs()
                 ->andReturn(\WPDesk\HttpClient\Curl\CurlClient::class);