Fix integration tests after changes in python interface
!122 (merged) introduced a few minor changes to the python interface that break some integration tests. This MR updates the integration tests to the new python interface.
!122 (merged) introduced a few minor changes to the python interface that break some integration tests. This MR updates the integration tests to the new python interface.