Fixes tests for python 3 (issue #3)