Fix issues with referencing main file