Fix return type of main.