We should probably pop this out into it's own test file.