lib/moneytree/payment_provider/base.rb in moneytree-rails-0.1.0 vs lib/moneytree/payment_provider/base.rb in moneytree-rails-0.1.1

@@ -1,9 +1,12 @@
 module Moneytree
   module PaymentProvider
     class Base
-      def initialize(account)
-        @account = account
+      attr_reader :payment_gateway, :account
+
+      def initialize(payment_gateway)
+        @payment_gateway = payment_gateway
+        @account = @payment_gateway.account
       end
     end
   end
 end