Bugfix connection lost visual

Not sure if this is the best way to do it, would like to discuss after you get a chance to look at it.