diff --git a/test/plugins/test_yapf_format.py b/test/plugins/test_yapf_format.py
index 202d7613..9bcc28f9 100644
--- a/test/plugins/test_yapf_format.py
+++ b/test/plugins/test_yapf_format.py
@@ -65,7 +65,7 @@ def test_config_file(tmpdir, workspace):
 
 def test_format_with_tab_size_option(workspace):
     doc = Document(DOC_URI, workspace, FOUR_SPACE_DOC)
-    res = pyls_format_document(doc, { "tabSize": "8" })
+    res = pylsp_format_document(doc, { "tabSize": "8" })
 
     assert len(res) == 1
     assert res[0]['newText'] == FOUR_SPACE_DOC.replace("    ", "        ")
@@ -73,7 +73,7 @@ def test_format_with_tab_size_option(workspace):
 
 def test_format_with_insert_spaces_option(workspace):
     doc = Document(DOC_URI, workspace, FOUR_SPACE_DOC)
-    res = pyls_format_document(doc, { "insertSpaces": False })
+    res = pylsp_format_document(doc, { "insertSpaces": False })
 
     assert len(res) == 1
     assert res[0]['newText'] == FOUR_SPACE_DOC.replace("    ", "\t")
@@ -81,7 +81,7 @@ def test_format_with_insert_spaces_option(workspace):
 
 def test_format_with_yapf_specific_option(workspace):
     doc = Document(DOC_URI, workspace, FOUR_SPACE_DOC)
-    res = pyls_format_document(doc, { "USE_TABS": True })
+    res = pylsp_format_document(doc, { "USE_TABS": True })
 
     assert len(res) == 1
     assert res[0]['newText'] == FOUR_SPACE_DOC.replace("    ", "\t")