Added association validation