fixed a thing in main