Fixed test failure maybe