Adjusted test section and set under main.