Fix main file references