Add support for passing `ok`, `nok` as separate states to attempts

1 files changed, 10 insertions, 2 deletions

diff --git a/src/construct/thematic_break.rs b/src/construct/thematic_break.rs
index 4ed25b6..30fabe4 100644
--- a/src/construct/thematic_break.rs
+++ b/src/construct/thematic_break.rs
@@ -72,7 +72,11 @@ pub fn start(tokenizer: &mut Tokenizer) -> State {
             },
         );
 
-        tokenizer.go(state_name, StateName::ThematicBreakBefore)
+        tokenizer.attempt(
+            state_name,
+            State::Fn(StateName::ThematicBreakBefore),
+            State::Nok,
+        )
     } else {
         State::Nok
     }
@@ -142,7 +146,11 @@ pub fn sequence(tokenizer: &mut Tokenizer) -> State {
         _ => {
             tokenizer.exit(Token::ThematicBreakSequence);
             let state_name = space_or_tab(tokenizer);
-            tokenizer.attempt_opt(state_name, StateName::ThematicBreakAtBreak)
+            tokenizer.attempt(
+                state_name,
+                State::Fn(StateName::ThematicBreakAtBreak),
+                State::Fn(StateName::ThematicBreakAtBreak),
+            )
         }
     }
 }