init: Update init to fix storages in KK