Fixes incorrect unit test