diff --git a/api/server/src/test/dataset/generate_graph.sh b/api/server/src/test/dataset/generate_graph.sh
index ffc7f07..d5aee4c 100755
--- a/api/server/src/test/dataset/generate_graph.sh
+++ b/api/server/src/test/dataset/generate_graph.sh
@@ -1,21 +1,22 @@
 #!/bin/bash
 
 # Build Docker work environment
 toplevel_dir=`git rev-parse --show-toplevel`
 mkdir -p dockerfiles
 cp -r $toplevel_dir/dockerfiles/ .
 docker build --tag swh-graph-test dockerfiles
 
 # Setup input for compression script
 tr ' ' '\n' < graph.edges.csv | sort -u > graph.nodes.csv
 gzip --force --keep graph.edges.csv
 gzip --force --keep graph.nodes.csv
 
 # Setup output
 rm -f stderr stdout
 
 docker run                                                      \
     --name swh-graph-test --rm --tty --interactive              \
     --volume $(pwd):/data swh-graph-test:latest                 \
     ./scripts/compress_graph.sh                                 \
-    --input /data/graph --output /data --lib /swh/graph-lib
+    --input /data/graph --output /data --lib /swh/graph-lib     \
+    --stdout /data/stdout --stderr /data/stderr
diff --git a/dockerfiles/scripts/compress_graph.sh b/dockerfiles/scripts/compress_graph.sh
index 7ad2168..fe43ef4 100755
--- a/dockerfiles/scripts/compress_graph.sh
+++ b/dockerfiles/scripts/compress_graph.sh
@@ -1,82 +1,87 @@
 #!/bin/bash
 
 usage() {
-    echo "Usage: --input <graph path> --output <out dir> --lib <graph lib path>"
+    echo "Usage: --input <graph path> --output <out dir> --lib <graph lib dir>"
     echo "  options:"
     echo "    -t, --tmp <temporary dir> (default to /tmp/)"
+    echo "    --stdout  <stdout file> (default to ./stdout)"
+    echo "    --stderr  <stderr file> (default to ./stderr)"
     exit 1
 }
 
 graph_path=""
 out_dir=""
-lib_path=""
+lib_dir=""
 tmp_dir="/tmp/"
+stdout_file="stdout"
+stderr_file="stderr"
 while (( "$#" )); do
     case "$1" in
         -i|--input) shift; graph_path=$1;;
         -o|--output) shift; out_dir=$1;;
-        -l|--lib) shift; lib_path=$1;;
+        -l|--lib) shift; lib_dir=$1;;
         -t|--tmp) shift; tmp_dir=$1;;
+        --stdout) shift; stdout_file=$1;;
+        --stderr) shift; stderr_file=$1;;
         *) usage;;
     esac
     shift
 done
 
-if [[ -z $graph_path || -z $out_dir || -z $lib_path ]]; then
+if [[ -z $graph_path || -z $out_dir || -z $lib_dir ]]; then
     usage
 fi
 
+if [[ -f "$stdout_file" || -f "$stderr_file" ]]; then
+    echo "Cannot overwrite previous compression stdout/stderr files"
+    exit 1
+fi
+
 dataset=$(basename $graph_path)
 compr_graph_path="$out_dir/$dataset"
-stdout_file="$out_dir/stdout"
-stderr_file="$out_dir/stderr"
 
 mkdir -p $out_dir
 mkdir -p $tmp_dir
-if [[ -f "$stdout_file" || -f "$stderr_file" ]]; then
-    echo "Cannot overwrite compression stdout/stderr files"
-    exit 1
-fi
 
 java_cmd () {
-    /usr/bin/time -v java -cp $lib_path/'*' $*
+    /usr/bin/time -v java -cp $lib_dir/'*' $*
 }
 
 {
     # Build a function (MPH) that maps node names to node numbers in
     # lexicographic order (output: .mph)
     java_cmd it.unimi.dsi.sux4j.mph.GOVMinimalPerfectHashFunction       \
         --zipped $compr_graph_path.mph --temp-dir $tmp_dir              \
         $graph_path.nodes.csv.gz                                        ;
 
     # Build the graph in BVGraph format (output: .{graph,offsets,properties})
     java_cmd it.unimi.dsi.big.webgraph.ScatteredArcsASCIIGraph          \
         --function $compr_graph_path.mph --temp-dir $tmp_dir            \
         --zipped $compr_graph_path-bv < $graph_path.edges.csv.gz        ;
     # Build the offset big-list file to load the graph faster (output: .obl)
     java_cmd it.unimi.dsi.big.webgraph.BVGraph                          \
         --list $compr_graph_path-bv                                     ;
 
     # Find a better permutation using a BFS traversal order (output: .order)
     java_cmd it.unimi.dsi.law.graph.BFSBig                              \
         $compr_graph_path-bv $compr_graph_path.order                    ;
 
     # Permute the graph accordingly
     batch_size=1000000000
     java_cmd it.unimi.dsi.big.webgraph.Transform mapOffline             \
         $compr_graph_path-bv $compr_graph_path                          \
         $compr_graph_path.order $batch_size                             ;
     java_cmd it.unimi.dsi.big.webgraph.BVGraph                          \
         --list $compr_graph_path                                        ;
 
     # Compute graph statistics (output: .{indegree,outdegree,stats})
     java_cmd it.unimi.dsi.big.webgraph.Stats $compr_graph_path          ;
 
     # Create transposed graph (to allow backward traversal)
     java_cmd it.unimi.dsi.big.webgraph.Transform transposeOffline       \
         $compr_graph_path $compr_graph_path-transposed $batch_size      ;
     java_cmd it.unimi.dsi.big.webgraph.BVGraph                          \
         --list $compr_graph_path-transposed                             ;
 } >> $stdout_file 2>> $stderr_file
 
 echo "Graph compression done."