Fixed test to match new style