Fix python bindings