Added the 'main' line in the unit test.