Fix brittle test broken by last commit