lib/devise-onesignal/device_concern.rb in devise-onesignal-3.0.0 vs lib/devise-onesignal/device_concern.rb in devise-onesignal-3.0.1

@@ -15,10 +15,10 @@
             unless cookies[:oneSignalPlayerId].nil?
                 onesignal_player_id = cookies[:oneSignalPlayerId]
                 device = Device.find_or_create_by! onesignal_id: onesignal_player_id
                 device.onesignal_permission = cookies[:oneSignalPlayerPermission]
                 device.last_used = Time.now
-                device.owner = set_device_owner
+                device.owner = ApplicationController.set_device_owner || set_device_owner
                 device.save!
             end
         end
 
         def set_device_owner