Remove unittest.main() call