Fixed a test that broke.