Correct test assertion