1 files changed, 9 insertions, 6 deletions
diff --git a/kvmd/plugins/msd/__init__.py b/kvmd/plugins/msd/__init__.py
index f43a40c2..95c0f0d0 100644
--- a/kvmd/plugins/msd/__init__.py
+++ b/kvmd/plugins/msd/__init__.py
@@ -27,6 +27,9 @@ from typing import Type
 from typing import AsyncGenerator
 from typing import Optional
 
+from ...errors import OperationError
+from ...errors import IsBusyError
+
 from .. import BasePlugin
 from .. import get_plugin_class
 
@@ -36,10 +39,15 @@ class MsdError(Exception):
     pass
 
 
-class MsdOperationError(MsdError):
+class MsdOperationError(OperationError, MsdError):
     pass
 
 
+class MsdIsBusyError(IsBusyError, MsdError):
+    def __init__(self) -> None:
+        super().__init__("Performing another MSD operation, please try again later")
+
+
 class MsdOfflineError(MsdOperationError):
     def __init__(self) -> None:
         super().__init__("MSD is not found")
@@ -70,11 +78,6 @@ class MsdImageExistsError(MsdOperationError):
         super().__init__("This image is already exists")
 
 
-class MsdIsBusyError(MsdOperationError):
-    def __init__(self) -> None:
-        super().__init__("Performing another MSD operation, please try again later")
-
-
 class MsdMultiNotSupported(MsdOperationError):
     def __init__(self) -> None:
         super().__init__("This MSD does not support storing multiple images")