Expand integration tests