Update to adopt ZRAM changes