[BUGFIX] Fix undefined variable in main()