small semantic fix in main()