Update for latest module API.