Fixing test to ensure it passes with python 3.