Fixed missing connection import in __init__