Fixed up this mornings tests.