Fixes test broken in commit 4e1fcfe