Minor update to the example to fix ui on connection fail