JS fix for menu