diff --git a/dvc_oss/tests/test_dvc.py b/dvc_oss/tests/test_dvc.py
index 2747d1d..62ce86c 100644
--- a/dvc_oss/tests/test_dvc.py
+++ b/dvc_oss/tests/test_dvc.py
@@ -1,6 +1,8 @@
 import pytest
-from dvc.testing.test_api import TestAPI  # noqa, pylint: disable=unused-import
-from dvc.testing.test_remote import (  # noqa, pylint: disable=unused-import
+from dvc.testing.api_tests import (  # noqa, pylint: disable=unused-import
+    TestAPI,
+)
+from dvc.testing.remote_tests import (  # noqa, pylint: disable=unused-import
     TestRemote,
 )