1 files changed, 3 insertions, 10 deletions
diff --git a/lib/sew_stack/stitch_layers/mixins/randomization.py b/lib/sew_stack/stitch_layers/mixins/randomization.py
index 5414731c..009b5042 100644
--- a/lib/sew_stack/stitch_layers/mixins/randomization.py
+++ b/lib/sew_stack/stitch_layers/mixins/randomization.py
@@ -1,17 +1,13 @@
 import os
 from secrets import randbelow
-from typing import TYPE_CHECKING
 
 import wx.propgrid
 
+from .protocol import LayerProtocol, with_protocol
 from ..stitch_layer_editor import Category, Property
 from ....i18n import _
 from ....svg import PIXELS_PER_MM
-from ....utils import DotDict, get_resource_dir, prng
-
-if TYPE_CHECKING:
-    from ... import SewStack
-
+from ....utils import get_resource_dir, prng
 
 editor_instance = None
 
@@ -70,10 +66,7 @@ class RandomizationPropertiesMixin:
         )
 
 
-class RandomizationMixin:
-    config: DotDict
-    element: "SewStack"
-
+class RandomizationMixin(with_protocol(LayerProtocol)):
     @classmethod
     def randomization_defaults(cls):
         return dict(