small fix in the main