Refactored out a missed jQuery call