diff --git a/openpype/tools/workfiles/files_widget.py b/openpype/tools/workfiles/files_widget.py
index 56af7752da..bb2ded3b94 100644
--- a/openpype/tools/workfiles/files_widget.py
+++ b/openpype/tools/workfiles/files_widget.py
@@ -18,6 +18,7 @@ from openpype.lib.avalon_context import (
     update_current_task,
     compute_session_changes
 )
+from openpype.pipeline import registered_host
 from .model import (
     WorkAreaFilesModel,
     PublishFilesModel,
@@ -93,7 +94,7 @@ class FilesWidget(QtWidgets.QWidget):
         # This is not root but workfile directory
         self._workfiles_root = None
         self._workdir_path = None
-        self.host = api.registered_host()
+        self.host = registered_host()
 
         # Whether to automatically select the latest modified
         # file on a refresh of the files model.
diff --git a/openpype/tools/workfiles/save_as_dialog.py b/openpype/tools/workfiles/save_as_dialog.py
index f5ae393d0f..0a7c7821ba 100644
--- a/openpype/tools/workfiles/save_as_dialog.py
+++ b/openpype/tools/workfiles/save_as_dialog.py
@@ -11,6 +11,7 @@ from openpype.lib import (
     get_last_workfile_with_version,
     get_workdir_data,
 )
+from openpype.pipeline import registered_host
 from openpype.tools.utils import PlaceholderLineEdit
 
 log = logging.getLogger(__name__)
@@ -65,7 +66,7 @@ class CommentMatcher(object):
             return
 
         # Create a regex group for extensions
-        extensions = api.registered_host().file_extensions()
+        extensions = registered_host().file_extensions()
         any_extension = "(?:{})".format(
             "|".join(re.escape(ext[1:]) for ext in extensions)
         )
@@ -200,7 +201,7 @@ class SaveAsDialog(QtWidgets.QDialog):
         self.setWindowFlags(self.windowFlags() | QtCore.Qt.FramelessWindowHint)
 
         self.result = None
-        self.host = api.registered_host()
+        self.host = registered_host()
         self.root = root
         self.work_file = None
         self._extensions = extensions