fix main.k