Fixed python 2 testing bug.