Fix test broken in gh-708