BUG: Fixed test