diff --git a/dockerfiles/ink-waterfall-ci/Dockerfile b/dockerfiles/ink-waterfall-ci/Dockerfile
index 21a1cee4..55303f57 100644
--- a/dockerfiles/ink-waterfall-ci/Dockerfile
+++ b/dockerfiles/ink-waterfall-ci/Dockerfile
@@ -66,7 +66,7 @@ RUN	set -eux; \
 	sed -i 's/name = "substrate-contracts-node"/name = "substrate-contracts-node-rand-extension"/g' substrate-contracts-node/node/Cargo.toml && \
 	cargo install --locked --path substrate-contracts-node/node/ && \
 # Needed for regression testing, a CSV contains the sizes of compiled contracts.
-	cargo install --git https://github.com/HCastano/csv-comparator && \
+	cargo install --git https://github.com/paritytech/ink-waterfall.git csv-comparator && \
 	npm install -g csv2md && \
 # versions
 	rustup show && \