should fix integration tests