minor fix in main