lib/cassanity/argument_generators/keyspace_use.rb in cassanity-0.3.0 vs lib/cassanity/argument_generators/keyspace_use.rb in cassanity-0.4.0

@@ -2,10 +2,10 @@
   module ArgumentGenerators
     class KeyspaceUse
 
       # Internal
       def call(args = {})
-        name = args.fetch(:name)
+        name = args.fetch(:keyspace_name)
         cql = "USE #{name}"
         [cql]
       end
     end
   end