Fixing tests for py32