Fix main()