- Quick bug fix in main.