Fixes for changes due to 'six' module