Fix the jQuery plugin