Fix the other instance of failing tests