diff --git a/lightly/models/modules/heads.py b/lightly/models/modules/heads.py
index 58d7b78f3..5deaf50a9 100644
--- a/lightly/models/modules/heads.py
+++ b/lightly/models/modules/heads.py
@@ -350,14 +350,14 @@ def __init__(
 
 
 class SMoGPrototypes(nn.Module):
-        """SMoG prototypes module for synchronous momentum grouping.
+    """SMoG prototypes module for synchronous momentum grouping.
 
-        Args:
-            group_features: 
-                Tensor containing the group features.
-            beta: 
-                Beta parameter for momentum updating.
-        """
+    Args:
+        group_features: 
+            Tensor containing the group features.
+        beta: 
+            Beta parameter for momentum updating.
+    """
 
     def __init__(
         self,