CONN-612 Updated unit test