small fix in main
Added coverage