diff --git a/Controller/Adminhtml/Config/Sftp/TestConnection.php b/Controller/Adminhtml/Config/Sftp/TestConnection.php
index 79dc2bb..ef0cd7f 100644
--- a/Controller/Adminhtml/Config/Sftp/TestConnection.php
+++ b/Controller/Adminhtml/Config/Sftp/TestConnection.php
@@ -14,6 +14,7 @@
use Magento\Store\Api\Data\StoreInterface;
use Magento\Store\Model\StoreManagerInterface;
use Magento\Store\Model\ScopeInterface;
+use Magento\Framework\Filesystem\Io\Sftp;
/**
* Class TestConnection
@@ -31,7 +32,7 @@ class TestConnection extends Action
*/
public function __construct(
Context $context,
- \Bazaarvoice\Connector\Model\Filesystem\Io\Sftp $sftp,
+ Sftp $sftp,
ConfigProviderInterface $configProvider,
StoreInterface $store,
StoreManagerInterface $storeManager,
diff --git a/Model/Feed/Feed.php b/Model/Feed/Feed.php
index dc953fa..891d6f7 100644
--- a/Model/Feed/Feed.php
+++ b/Model/Feed/Feed.php
@@ -10,7 +10,7 @@
use Bazaarvoice\Connector\Api\ConfigProviderInterface;
use Bazaarvoice\Connector\Api\StringFormatterInterface;
-use Bazaarvoice\Connector\Model\Filesystem\Io\Sftp;
+use Magento\Framework\Filesystem\Io\Sftp;
use Bazaarvoice\Connector\Model\Source\Scope;
use Exception;
use Magento\Store\Model\Group;
diff --git a/etc/di.xml b/etc/di.xml
index 9193807..c207404 100644
--- a/etc/di.xml
+++ b/etc/di.xml
@@ -44,10 +44,6 @@
-
-
-
-