From ea4758e44201507d63b949bbea243909d3fd15f6 Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Sun, 25 Dec 2016 14:41:06 -0700
Subject: [PATCH] Clean up end of next()

---
 src/lexer/mod.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/lexer/mod.rs b/src/lexer/mod.rs
index 5aedd1b..24476a8 100644
--- a/src/lexer/mod.rs
+++ b/src/lexer/mod.rs
@@ -167,11 +167,10 @@ impl Iterator for Lexer {
             }
         }
         self.advance_begin();
-        let mut lex: Option<Lex> = None;
-        if let Some(token) = token {
-            lex = Some(Lex::new(token));
+        match token {
+            Some(t) => Some(Lex::new(t)),
+            None => None,
         }
-        lex
     }
 }