Worth adding integration test for if not already there.