Fix issue with TOC expanding