Oops. Fixed the tests now.