Fixed test assertions