Add correct tests.