Wrote some better integration tests.