Fixing TOC issue