Fixed the bug in main, again