Fix js on desktop for nav