Removed unnecessary code from unittests main