Fixed those small connection errors.

Fixed those small connection errors.