Woops... fix tests.