fixing main ..