Small test simplification