diff --git a/lib/pundit.rb b/lib/pundit.rb
index fd7127e..a147417 100644
--- a/lib/pundit.rb
+++ b/lib/pundit.rb
@@ -83,9 +83,12 @@ module Pundit
   attr_writer :policy_scope
 
   def policy(record)
-    @policy or Pundit.policy!(pundit_user, record, self.class.parent)
+    @_policy or Pundit.policy!(pundit_user, record, self.class.parent)
+  end
+
+  def policy=(policy)
+    @_policy = policy
   end
-  attr_writer :policy
 
   def pundit_user
     current_user