Update init logic