Fixed another bug in main