Make it work with pyside