diff --git a/src/java/com/datomic/CassandraCAS.java b/src/java/com/datomic/CassandraCAS.java
index 3090d47..e7a5163 100644
--- a/src/java/com/datomic/CassandraCAS.java
+++ b/src/java/com/datomic/CassandraCAS.java
@@ -48,19 +48,19 @@ public static class ThreeRetryPolicy implements RetryPolicy {
 
         @Override
         public RetryDecision onReadTimeout(Statement statement, ConsistencyLevel cl, int i, int i1, boolean b, int retries) {
-            return (retries == 3) ? RetryDecision.rethrow() : RetryDecision.retry(cl);
+            return (retries == 3) ? RetryDecision.rethrow() : RetryDecision.retry(statement.getConsistencyLevel());
         }
 
         @Override
         public RetryDecision onWriteTimeout(Statement statement, ConsistencyLevel cl, WriteType writeType, int i, int i1, int retries) {
             System.out.println(retries);
             System.out.println(cl);
-            return (retries == 3) ? RetryDecision.rethrow() : RetryDecision.retry(cl);
+            return (retries == 3) ? RetryDecision.rethrow() : RetryDecision.retry(statement.getConsistencyLevel());
         }
 
         @Override
         public RetryDecision onUnavailable(Statement statement, ConsistencyLevel cl, int i, int i1, int retries) {
-            return (retries == 3) ? RetryDecision.rethrow() : RetryDecision.retry(cl);
+            return (retries == 3) ? RetryDecision.rethrow() : RetryDecision.retry(statement.getConsistencyLevel());
         }
     }
     public static final RetryPolicy retryPolicy = new ThreeRetryPolicy();
@@ -177,6 +177,5 @@ public static void main(String[] args) throws IOException {
             driver.createRev();
             driver.race(Integer.parseInt(args[3]), Integer.parseInt(args[4]), Long.parseLong(args[5]));
         }
-        driver.close();
     }
 }