Fix: Register message event listener on init