diff --git a/source/MaterialXTest/MaterialXCore/Document.cpp b/source/MaterialXTest/MaterialXCore/Document.cpp
index 853bfaf1cf..94f8b98708 100644
--- a/source/MaterialXTest/MaterialXCore/Document.cpp
+++ b/source/MaterialXTest/MaterialXCore/Document.cpp
@@ -105,8 +105,12 @@ TEST_CASE("Document", "[document]")
     customImpl->setNodeDef(customNodeDef);
     REQUIRE(customLibrary->validate());
 
-    // Set data library reference.
-    doc->setDataLibrary(customLibrary);
+    // Import the custom library.
+    mx::DocumentPtr customDatalibrary = mx::createDocument();
+    customDatalibrary->importLibrary(customLibrary);
+
+    // Set data library
+    doc->setDataLibrary(customDatalibrary);
     mx::NodeGraphPtr importedNodeGraph = doc->getNodeGraph("custom:NG_custom");
     mx::NodeDefPtr importedNodeDef = doc->getNodeDef("custom:ND_simpleSrf");
     mx::ImplementationPtr importedImpl = doc->getImplementation("custom:IM_custom");