diff --git a/LabsSolutions/02-pytorch-asr/models.py b/LabsSolutions/02-pytorch-asr/models.py
index 552c7c1..5e02d73 100644
--- a/LabsSolutions/02-pytorch-asr/models.py
+++ b/LabsSolutions/02-pytorch-asr/models.py
@@ -219,8 +219,7 @@ def beam_decode(self, inputs: PackedSequence, beam_size: int, blank_id: int):
             outputs = self.forward(inputs)
         return beam_decode(outputs, beam_size, blank_id, self.charmap)
 
-
-# @SOL
+    # SOL@
 
 
 class CTCModel(nn.Module):
@@ -473,6 +472,8 @@ def beam_decode(self, inputs: PackedSequence, beam_size: int, blank_id: int):
             outputs = self.forward(inputs)
         return beam_decode(outputs, beam_size, blank_id, self.charmap)
 
+    # SOL@
+
 
 # @SOL
 def ex_ctc():