Fixes another test error