Fixed error in main