diff --git a/core/leras/layers/__init__.py b/core/leras/layers/__init__.py
index 1c81963..d8f1c9d 100644
--- a/core/leras/layers/__init__.py
+++ b/core/leras/layers/__init__.py
@@ -13,4 +13,5 @@ from .FRNorm2D import *
 from .TLU import *
 from .ScaleAdd import *
 from .DenseNorm import *
-from .AdaIN import *
\ No newline at end of file
+from .AdaIN import *
+from .MsSsim import *