1 files changed, 6 insertions, 1 deletions
diff --git a/ccl_test.go b/ccl_test.go
index ab4833f..53b2a24 100644
--- a/ccl_test.go
+++ b/ccl_test.go
@@ -684,8 +684,13 @@ func BenchmarkLex(b *testing.B) {
 		repeated: [5, 6]
 	`)
 	for b.Loop() {
-		for _, err := range tokens(msg) {
+		l := newLexer(msg)
+		for {
+			_, err := l.next()
 			if err != nil {
+				if err == errEOF {
+					break
+				}
 				b.Fatal(err)
 			}
 		}