Add test for association