Make it work also when toc is enabled