diff --git a/app/models/order_decorator.rb b/app/models/order_decorator.rb
index 5c7edab..d634d70 100644
--- a/app/models/order_decorator.rb
+++ b/app/models/order_decorator.rb
@@ -1,4 +1,4 @@
-Order.class_eval do
+Spree::Order.class_eval do
   
   # Are all products/variants of this Order to be downloaded by the customer?
   def digital?
@@ -23,4 +23,4 @@ Order.class_eval do
     nil
   end
 
-end
\ No newline at end of file
+end
diff --git a/app/models/variant_decorator.rb b/app/models/variant_decorator.rb
index bf07ceb..acf8eb5 100644
--- a/app/models/variant_decorator.rb
+++ b/app/models/variant_decorator.rb
@@ -1,4 +1,4 @@
-Variant.class_eval do
+Spree::Variant.class_eval do
   
   has_one :digital, :dependent => :destroy
   after_save :destroy_digital, :if => :deleted?
@@ -17,4 +17,4 @@ Variant.class_eval do
     digital.destroy
   end
 
-end
\ No newline at end of file
+end