diff --git a/lib/python/ray/worker.py b/lib/python/ray/worker.py
index 0f84c34f8..afc7c84cd 100644
--- a/lib/python/ray/worker.py
+++ b/lib/python/ray/worker.py
@@ -339,7 +339,7 @@ class Worker(object):
     """
     if ray.lib.is_arrow(self.handle, objref):
       ## this is the new codepath
-      buff, segmentid, metadata_offset = ray.libraylib.get_buffer(self.handle, objref)
+      buff, segmentid, metadata_offset = ray.lib.get_buffer(self.handle, objref)
       metadata_size = np.frombuffer(buff, dtype="int64", count=1)[0]
       metadata = np.frombuffer(buff, dtype="byte", offset=8, count=metadata_size)
       data = np.frombuffer(buff, dtype="byte")[8 + metadata_size:]
diff --git a/test/runtest.py b/test/runtest.py
index 0f23c5e2f..0a783037c 100644
--- a/test/runtest.py
+++ b/test/runtest.py
@@ -137,10 +137,10 @@ class ObjStoreTest(unittest.TestCase):
 
     # Getting a buffer after modifying it before it finishes should return updated buffer
     objref =  ray.lib.get_objref(w1.handle)
-    buf = ray.libraylib.allocate_buffer(w1.handle, objref, 100)
+    buf = ray.lib.allocate_buffer(w1.handle, objref, 100)
     buf[0][0] = 1
-    ray.libraylib.finish_buffer(w1.handle, objref, buf[1], 0)
-    completedbuffer = ray.libraylib.get_buffer(w1.handle, objref)
+    ray.lib.finish_buffer(w1.handle, objref, buf[1], 0)
+    completedbuffer = ray.lib.get_buffer(w1.handle, objref)
     self.assertEqual(completedbuffer[0][0], 1)
 
     ray.services.cleanup()