diff --git a/l2/3.lex b/l2/3.lex
index 192279d..b6ed871 100644
--- a/l2/3.lex
+++ b/l2/3.lex
@@ -2,6 +2,7 @@
     #include <stdbool.h>
 
     bool leave_docs = false;
+    bool in_docs = false;
 %}
 
 %x SINGLE_LINE
@@ -19,32 +20,54 @@
 <STRING>.               { ECHO; }
 
 <INITIAL>"///"|"//!" {
+    in_docs = true;
     if (leave_docs) {
         ECHO;
-    } else {
-        BEGIN(SINGLE_LINE);
     }
+    BEGIN(SINGLE_LINE);
 }
 
 <INITIAL>"/**/"         { putchar(' '); }
 
 <INITIAL>"/**"|"/*!" {
+    in_docs = true;
     if (leave_docs) {
         ECHO;
-    } else {
-        BEGIN(MULTI_LINE);
     }
+    BEGIN(MULTI_LINE);
 }
 
 <INITIAL>"//"           { BEGIN(SINGLE_LINE); }
 <INITIAL>.              { ECHO; }
 <SINGLE_LINE>"\\\n"     { }
-<SINGLE_LINE>\n         { BEGIN(INITIAL); }
-<SINGLE_LINE>.          { }
+<SINGLE_LINE>\n {
+    if (leave_docs && in_docs) {
+        ECHO;
+        in_docs = false;
+    }
+    BEGIN(INITIAL);
+}
+<SINGLE_LINE>. {
+    if (leave_docs && in_docs) {
+        ECHO;
+    }
+}
 
 <INITIAL>"/*"           { BEGIN(MULTI_LINE); }
-<MULTI_LINE>"*/"        { putchar(' '); BEGIN(INITIAL); }
-<MULTI_LINE>.           { }
+<MULTI_LINE>"*/" {
+    if (leave_docs && in_docs) {
+        ECHO;
+        in_docs = false;
+    } else {
+        putchar(' ');
+    }
+    BEGIN(INITIAL);
+}
+<MULTI_LINE>. {
+    if (leave_docs && in_docs) {
+        ECHO;
+    }
+}
 %%
 
 // single line comment