Fixing a dependency on 'six'