Fixed one test to work better.