diff --git a/build/Snakefile-finish b/build/Snakefile-finish
index 66dc331f..7fc188c9 100644
--- a/build/Snakefile-finish
+++ b/build/Snakefile-finish
@@ -1,6 +1,6 @@
 rule Finish:
     input:
-        code = config['BUILD_CODE_DIR'] + "/finish-snakemake.sh,
+        code = config['FINISH_SCRIPT'],
         merged_output_nodes_file = config['MERGED_OUTPUT_NODES_FILE'],
         final_output_edges_file = config['MERGED_OUTPUT_EDGES_FILE'],
         output_file_orphan_edges = config['OUTPUT_FILE_ORPHAN_EDGES'],
diff --git a/build/snakemake-config-var.yaml b/build/snakemake-config-var.yaml
index f4cce9ad..e530c7d9 100644
--- a/build/snakemake-config-var.yaml
+++ b/build/snakemake-config-var.yaml
@@ -275,4 +275,6 @@ kg2_tsv_dir: ${BUILD_DIR}/TSV
 kg2_tsv_tarball: ${BUILD_DIR}/kg2-tsv-for-neo4j${test_suffix}.tar.gz
 tsv_placeholder: ${BUILD_DIR}/tsv_placeholder.empty
 
+finish_script: ${BUILD_CODE_DIR}/finish-snakemake.sh
+
 version_file: ${BUILD_DIR}/kg2-version.txt