There may be a better way to write these tests.