1 files changed, 10 insertions, 4 deletions
diff --git a/tests/export-graph.nix b/tests/export-graph.nix
index b7afa2a85b9b..d65ad8c143cb 100644
--- a/tests/export-graph.nix
+++ b/tests/export-graph.nix
@@ -2,21 +2,27 @@ with import ./config.nix;
 
 rec {
 
-  buildGraphBuilder = builtins.toFile "build-graph-builder"
+  printRefs =
     ''
       while read path; do
           read drv
           read nrRefs
           echo "$path has $nrRefs references"
-          echo -n "$path" >> $out
+          echo "$path" >> $out
           for ((n = 0; n < $nrRefs; n++)); do read ref; echo "ref $ref"; done
       done < refs
     '';
 
-  buildGraph = mkDerivation {
+  runtimeGraph = mkDerivation {
     name = "dependencies";
-    builder = buildGraphBuilder;
+    builder = builtins.toFile "build-graph-builder" "${printRefs}";
     exportReferencesGraph = ["refs" (import ./dependencies.nix)];
   };
 
+  buildGraph = mkDerivation {
+    name = "dependencies";
+    builder = builtins.toFile "build-graph-builder" "${printRefs}";
+    exportBuildReferencesGraph = ["refs" (import ./dependencies.nix).drvPath];
+  };
+
 }