Fixed a test broken during earlier refactoring.