diff --git a/test/advanced-tocs/sec3.tex b/test/advanced-tocs/chapters.tex
similarity index 60%
rename from test/advanced-tocs/sec3.tex
rename to test/advanced-tocs/chapters.tex
index 09988cd..e31b051 100644
--- a/test/advanced-tocs/sec3.tex
+++ b/test/advanced-tocs/chapters.tex
@@ -1,3 +1,19 @@
+\chapter{Test chapter 1}
+This test document is created to test advanced TOC functionality.
+
+\section{Test section 1}
+
+\subsection{Test sub section 1}
+
+\section{$\alpha$}
+
+\subsection{Test sub section 1}
+
+\subsection{Test sub section 2}
+Some text.
+
+\chapter{Test chapter 2}
+
 \section{Testing TOC levels}
 Testing more input and TOC levels.
 
diff --git a/test/advanced-tocs/main.tex b/test/advanced-tocs/main.tex
index 5221bc8..2fe13e4 100644
--- a/test/advanced-tocs/main.tex
+++ b/test/advanced-tocs/main.tex
@@ -1,42 +1,28 @@
-\documentclass{scrartcl}
-
-\setcounter{secnumdepth}{2}
+\documentclass{book}
 
 \begin{document}
-This test document is created to test advanced TOC functionality.
 
-\section{Test section 1}
+\frontmatter
+\input{preface}
 
-\subsection{Test sub section 1}
-
-\section{$\alpha$}
-
-\input{sec2.tex}
+\mainmatter
+\input{chapters}
 
 \appendix
+\input{appendix}
 
-\input{sec3}
-
-% Test matching of bibliography additions
+\backmatter
 \bibliography{test}
-
 \printbibliography
-
 \printbibliography{Testing}
-
 \printbibliography[Some option]{Testing}
-
 \printbibheading
-
 \printbibheading{Testing}
-
 \printbibheading[Some option]{Testing}
-
 \begin{thebibliography}
 \end{thebibliography}
 
 \printindex
-
 \printindex[Some option]
 
 \end{document}
diff --git a/test/advanced-tocs/preface.tex b/test/advanced-tocs/preface.tex
new file mode 100644
index 0000000..67295cf
--- /dev/null
+++ b/test/advanced-tocs/preface.tex
@@ -0,0 +1,2 @@
+\chapter{Preface}
+Preface.
diff --git a/test/advanced-tocs/sec2.tex b/test/advanced-tocs/sec2.tex
deleted file mode 100644
index 77bb989..0000000
--- a/test/advanced-tocs/sec2.tex
+++ /dev/null
@@ -1,2 +0,0 @@
-\subsection{Test sub section 2}
-Some text.