Fixes a test error