Fixed callbacks when using association.build