diff --git a/py/selenium/webdriver/chromium/options.py b/py/selenium/webdriver/chromium/options.py
index 20a07dae97927..6a2ac1f8a164b 100644
--- a/py/selenium/webdriver/chromium/options.py
+++ b/py/selenium/webdriver/chromium/options.py
@@ -18,7 +18,7 @@
 import base64
 import os
 
-from selenium.webdriver import DesiredCapabilities
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
 from selenium.webdriver.common.options import ArgOptions
 
 
diff --git a/py/test/unit/selenium/webdriver/chrome/chrome_options_tests.py b/py/test/unit/selenium/webdriver/chrome/chrome_options_tests.py
index f14f99684717c..ff7e14ebeb84e 100644
--- a/py/test/unit/selenium/webdriver/chrome/chrome_options_tests.py
+++ b/py/test/unit/selenium/webdriver/chrome/chrome_options_tests.py
@@ -92,7 +92,7 @@ def test_add_encoded_extension(options):
 def test_get_extensions_from_extension_files(options, mocker):
     null = 'NUL' if platform.system().lower() == 'windows' else '/dev/null'
     mocker.patch(
-        'selenium.webdriver.chrome.options.open'.format(__name__)).return_value = open(null)
+        'selenium.webdriver.chromium.options.open'.format(__name__)).return_value = open(null)
     mocker.patch('base64.b64encode').return_value = 'foo'.encode()
     options._extension_files = ['foo']
     assert 'foo' in options.extensions