Added check connection to test