Another minor fix in main.