1 files changed, 4 insertions, 1 deletions
diff --git a/lib/parse/lex.js b/lib/parse/lex.js
index 6b3236e..def74b4 100644
--- a/lib/parse/lex.js
+++ b/lib/parse/lex.js
@@ -23,7 +23,10 @@ function sectionType(nodes, idx) {
         type: 'code'
     }).length;
 
-    if(codeNodes === 3 && (idx % 2) == 1) {
+    if(
+        (codeNodes === 3 || codeNodes === 4) &&
+        (idx % 2) === 1)
+    {
         return 'exercise';
     }