Fix main.ml