very common exceptions

1 files changed, 5 insertions, 2 deletions

diff --git a/kvmd/plugins/atx/__init__.py b/kvmd/plugins/atx/__init__.py
index a593929e..b536ef87 100644
--- a/kvmd/plugins/atx/__init__.py
+++ b/kvmd/plugins/atx/__init__.py
@@ -24,6 +24,9 @@ from typing import Dict
 from typing import AsyncGenerator
 from typing import Type
 
+from ...errors import OperationError
+from ...errors import IsBusyError
+
 from .. import BasePlugin
 from .. import get_plugin_class
 
@@ -33,11 +36,11 @@ class AtxError(Exception):
     pass
 
 
-class AtxOperationError(AtxError):
+class AtxOperationError(OperationError, AtxError):
     pass
 
 
-class AtxIsBusyError(AtxOperationError):
+class AtxIsBusyError(IsBusyError, AtxError):
     def __init__(self) -> None:
         super().__init__("Performing another ATX operation, please try again later")