diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py index 4b3c5381..d47d1d20 100644 --- a/tests/integration/__init__.py +++ b/tests/integration/__init__.py @@ -119,9 +119,9 @@ else: log.info('Using Cassandra version: %s', CASSANDRA_VERSION) CCM_KWARGS['version'] = CASSANDRA_VERSION -if CASSANDRA_VERSION > '2.1': +if CASSANDRA_VERSION >= '2.1': default_protocol_version = 3 -elif CASSANDRA_VERSION > '2.0': +elif CASSANDRA_VERSION >= '2.0': default_protocol_version = 2 else: default_protocol_version = 1