Fix onInit