Minor tweak to try reconnecting