Added missing wait_for_ajax