Switch from AJS.onInit to AJS.$