Added listening for messages as explicit call to avoid discarding messages during setup.