Fix test to account for new composition