Fix missing import for ConnectionError