Add missing namespace imports in tests.