Correct connection

* Correct connection with type hinting