lib/octopus/association.rb in ar-octopus-0.0.18 vs lib/octopus/association.rb in ar-octopus-0.0.19

@@ -72,10 +72,10 @@
         return_val = association.send(constructor, attributees, replace_existing)
       else
         return_val = association.send(constructor, attributees)
       end
 
-      if have_a_valid_shard?
+      if should_set_current_shard?
         return_val.current_shard = self.current_shard
       end
 
       return_val
     end