Fixing associations tests by using andThen