fixed bug in main