Implement connection exception
fix test