Fix test to match change in code. :knucklerap: