Fixed a test broken by previous commit