Fix the bug for the TestConnection