Fixed borken tests.