Fix main to point to the correct file