Remove debug prints and fix integration tests