Used active record associations to make tests pass