Started implementing an integration test, will have to revisit.