Updating tests to be compatible with older python