Add missing os import to tests.