Fix jQuery selectors

Fallout from using different function before