diff --git a/libs/partners/openai/tests/unit_tests/test_load.py b/libs/partners/openai/tests/unit_tests/test_load.py
index d25b00a0b87..b1aabd07ccc 100644
--- a/libs/partners/openai/tests/unit_tests/test_load.py
+++ b/libs/partners/openai/tests/unit_tests/test_load.py
@@ -1,5 +1,4 @@
-from langchain_core.load.dump import dumpd, dumps
-from langchain_core.load.load import load, loads
+from langchain_core.load import dumpd, dumps, load, loads
 
 from langchain_openai import ChatOpenAI, OpenAI