small fix in main