Fix class section :bug: