lib/parametric/block_validator.rb in parametric-0.2.9 vs lib/parametric/block_validator.rb in parametric-0.2.10

@@ -6,30 +6,30 @@
       klass
     end
 
     def self.message(&block)
       @message_block = block if block_given?
-      @message_block
+      @message_block if instance_variable_defined?('@message_block')
     end
 
     def self.validate(&validate_block)
       @validate_block = validate_block if block_given?
-      @validate_block
+      @validate_block if instance_variable_defined?('@validate_block')
     end
 
     def self.coerce(&coerce_block)
       @coerce_block = coerce_block if block_given?
       @coerce_block
     end
 
     def self.eligible(&block)
       @eligible_block = block if block_given?
-      @eligible_block
+      @eligible_block if instance_variable_defined?('@eligible_block')
     end
 
     def self.meta_data(&block)
       @meta_data_block = block if block_given?
-      @meta_data_block
+      @meta_data_block if instance_variable_defined?('@meta_data_block')
     end
 
     attr_reader :message
 
     def initialize(*args)