Is it necessary to replace `module_state_key` with `module_id` in this file?  Isn't there also a defined property (that was introduced) called `module_state_key`?