Fixed an issue with the Main JS file