diff --git a/private/source.rkt b/private/source.rkt
index b68c490..a16631e 100644
--- a/private/source.rkt
+++ b/private/source.rkt
@@ -201,6 +201,7 @@
 (define (source-tokens src)
   (with-input-from-source src
     (λ ()
+      (port-count-lines! (current-input-port))
       (define tokens (make-vector-builder))
       (let loop ([offset 0] [mode #false])
         (define-values (text raw-attributes delimiter-kind start end _ new-mode)