Fixed an issue that broke a test.