Fix imports in python2
Add missing (object)