diff --git a/Dockerfile b/Dockerfile
index 69d01257..5456b48c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,11 +10,11 @@ COPY --chown=opam:root mirage/config.ml /home/opam/www/mirage/
 COPY --chown=opam:root mirageio.opam /home/opam/www/
 ARG TARGET=unix
 ARG EXTRA_FLAGS=
-ARG EXTRA_FLAGS_NO_METRICS="--tls=true --separate-networks"
 RUN opam pin add -n ocaml-solo5 'https://github.com/shym/ocaml-solo5.git#ocaml-5.2-reb'
-RUN opam exec -- mirage configure -f mirage/config.ml -t $TARGET $EXTRA_FLAGS_NO_METRICS
+RUN opam pin add -n memtrace-mirage 'https://github.com/Firobe/memtrace-mirage.git'
+RUN opam exec -- mirage configure -f mirage/config.ml -t $TARGET $EXTRA_ARGS
 RUN opam exec -- make depend
 COPY --chown=opam:root . /home/opam/www
-RUN opam exec -- mirage configure -f mirage/config.ml -t $TARGET $EXTRA_FLAGS_NO_METRICS
+RUN opam exec -- mirage configure -f mirage/config.ml -t $TARGET $EXTRA_ARGS
 RUN opam exec -- dune build mirage/ --profile release
 RUN if [ $TARGET = hvt ]; then sudo cp mirage/dist/www.$TARGET /unikernel.$TARGET; fi