Fixed test setup