fixed issue with test section.