Fix for custom message registry access