lib/ronin/ui/command_line/commands/payload.rb in ronin-exploits-0.2.0 vs lib/ronin/ui/command_line/commands/payload.rb in ronin-exploits-0.2.1

@@ -75,18 +75,18 @@
               Build the specified payload
             }
           end
 
           def arguments(*args)
-            Database.setup!
+            Database.setup
 
             # Load the payload
             if @path
               payload = Payloads::Payload.load_from(@path)
             elsif args.length >= 1
               @query[:name] = args.first
 
-              unless (payload = Payloads::Payload.first(@query))
+              unless (payload = Payloads::Payload.load_first(@query))
                 fail("could not find the specified payload")
               end
             else
               fail("must either specify a payload NAME or a PATH")
             end