Fix another issue w/ new tests.