diff --git a/lib/ProMotion/product.rb b/lib/ProMotion/product.rb
index 32536fa..cc615e3 100644
--- a/lib/ProMotion/product.rb
+++ b/lib/ProMotion/product.rb
@@ -20,8 +20,9 @@ def purchase(&callback)
 
     def restore(&callback)
       restore_iaps(product_id) do |status, products|
-        callback.call status, products.find{|p| p[:product_id] == product_id }
+        product = products.is_a?(Hash) ? products : products.find{|p| p[:product_id] == product_id }
+        callback.call status, product
       end
     end
   end
-end
\ No newline at end of file
+end