Bugfix unit test

It was incompatible with python 2.6