Fixed a test for python < 2.7