Fix segment register tests