correcing integration tests