diff --git a/src/cli.cpp b/src/cli.cpp
index 44118e7..2688e3a 100644
--- a/src/cli.cpp
+++ b/src/cli.cpp
@@ -8,6 +8,7 @@
 #include <boost/process.hpp>
 
 #include "aktualizr-lite/api.h"
+#include "aktualizr-lite/aklite_client_ext.h"
 #include "aktualizr-lite/cli/cli.h"
 #include "cli.h"
 
@@ -112,7 +113,7 @@ static void print_status(aklite::cli::StatusCode ret) {
   std::cout << std::endl;
 }
 
-static std::unique_ptr<AkliteClient> init_client(bool online_mode) {
+static std::unique_ptr<AkliteClientExt> init_client(bool online_mode) {
   boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::info);
 
   std::vector<boost::filesystem::path> cfg_dirs;
@@ -132,7 +133,7 @@ static std::unique_ptr<AkliteClient> init_client(bool online_mode) {
   }
 
   try {
-    return std::make_unique<AkliteClient>(cfg_dirs, false, false);
+    return std::make_unique<AkliteClientExt>(cfg_dirs, false, false);
   } catch (const std::exception& exc) {
     LOG_ERROR << "Failed to initialize the client: " << exc.what();
     return nullptr;