Fix AJAX callback issue