examples/keyspaces.rb in cassanity-0.2.1 vs examples/keyspaces.rb in cassanity-0.2.2

@@ -21,10 +21,10 @@
     replication_factor: 1,
   },
 })
 
 # drop keyspace if it exists
-keyspace.drop if connection.keyspace?('cassanity_examples')
+keyspace.drop if keyspace.exists?
 
 # create the keyspace, uses options from above
 keyspace.create
 
 # use this keyspace