fix little bug in main