diff --git a/test/main-test.tex b/test/main-test.tex
index d76ff18..d1ce348 100644
--- a/test/main-test.tex
+++ b/test/main-test.tex
@@ -468,8 +468,6 @@
 
 \input data-lines
 
-\ifcontext{\stoptext}
-\iflatex{
-  \def\crfnmtheend{\end{document}}
-  \expandafter\crfnmtheend
-}
\ No newline at end of file
+\ifcontext{\let\crfnmtheend\stoptext}
+\iflatex{\def\crfnmtheend{\end{document}}}
+\crfnmtheend