Update to match new module form.