diff --git a/tests/unit-tests/gpartition/conftest.py b/tests/unit-tests/gpartition/conftest.py
index 9f2770e1c7..9d1cfa7dfd 100644
--- a/tests/unit-tests/gpartition/conftest.py
+++ b/tests/unit-tests/gpartition/conftest.py
@@ -21,8 +21,7 @@
 import pytest
 
 from graphstorm.gpartition import LocalPartitionAlgorithm
-from graphstorm.gpartition.post_hard_negative import (shuffle_hard_negative_nids,
-                                                      load_hard_negative_config)
+
 
 @pytest.fixture(scope="module", name="chunked_metadata_dict")
 def metadata_dict_fixture() -> Dict: