lib/rapnd/notification.rb in rapnd-0.1.2 vs lib/rapnd/notification.rb in rapnd-0.1.3

@@ -1,13 +1,12 @@
 module Rapnd
   class Notification
     attr_accessor :badge, :alert, :sound, :custom, :device_token
     
     def initialize(hash)
-      [:badge, :alert, :sound, :device_token].each do |k|
-        self.instance_variable_set("@#{k}".to_sym, hash.delete(k)) if hash[k]
+      [:badge, :alert, :sound, :device_token, :custom].each do |k|
+        self.instance_variable_set("@#{k}".to_sym, hash[k]) if hash[k]
       end
-      @custom = hash
       raise 'Must provide device token' if self.device_token.nil?
       self.device_token = self.device_token.delete(' ')
     end
     
     def payload
\ No newline at end of file