added associations to make tests pass