Fixed a build error in main.