fixes test assertion for python 2