Work around circular import problems in helper scripts
Fix missing import for ConnectionError