Fixes yet another test error