diff --git a/python/legate_dataframe/lib/core/data_type.pyx b/python/legate_dataframe/lib/core/data_type.pyx
index fab5676..fc12789 100644
--- a/python/legate_dataframe/lib/core/data_type.pyx
+++ b/python/legate_dataframe/lib/core/data_type.pyx
@@ -10,7 +10,7 @@ from pylibcudf.types cimport DataType
 from pylibcudf.types cimport data_type as cpp_cudf_type
 
 import cudf
-from cudf._lib import pylibcudf
+import pylibcudf
 from cudf._lib.types import PYLIBCUDF_TO_SUPPORTED_NUMPY_TYPES, dtype_to_pylibcudf_type
 from numpy.typing import DTypeLike
 
diff --git a/python/tests/test_binaryop.py b/python/tests/test_binaryop.py
index b36d9e6..2a350e7 100644
--- a/python/tests/test_binaryop.py
+++ b/python/tests/test_binaryop.py
@@ -86,11 +86,11 @@ def test_scalar_input(cudf_column, op, scalar):
 
     res = binary_operation(col, scalar, op, cudf_column.dtype)
     expect = cudf_binaryop(cudf_column, cudf_scalar, op_str, cudf_column.dtype)
-    assert_frame_equal(res, cudf.Series(expect))
+    assert_frame_equal(res, expect)
 
     res = binary_operation(scalar, col, op, cudf_column.dtype)
     expect = cudf_binaryop(cudf_scalar, cudf_column, op_str, cudf_column.dtype)
-    assert_frame_equal(res, cudf.Series(expect))
+    assert_frame_equal(res, expect)
 
     with pytest.raises(ValueError, match="both inputs cannot be scalars"):
         binary_operation(scalar, scalar, op, cudf_column.dtype)