Fixed import for connection