small test simplification.
Fix test in python 3