Doh, fix test broken by last commit