Fix issue introduced in recent test fix