diff --git a/docs/storage-transfer.md b/docs/storage-transfer.md
index 1ef8f7a..c161a2d 100644
--- a/docs/storage-transfer.md
+++ b/docs/storage-transfer.md
@@ -101,7 +101,7 @@ The containers run with the following settings:
 
 | Container | Setting | Value |
 |---|---|---|
-| `imds-emulator` | Image | built from `imds-emulator/Dockerfile` |
+| `imds-emulator` | Image | built from `sts-agent/Dockerfile` |
 | | Mount | `/etc/sts-agent`, read only |
 | | Environment | `GOOGLE_APPLICATION_CREDENTIALS=/etc/sts-agent/cred.json` |
 | | Environment | `GOOGLE_API_CERTIFICATE_CONFIG=/etc/sts-agent/certificate_config.json` |
diff --git a/imds-emulator/Dockerfile b/sts-agent/Dockerfile
similarity index 100%
rename from imds-emulator/Dockerfile
rename to sts-agent/Dockerfile
diff --git a/sts-agent/compose.yaml b/sts-agent/compose.yaml
index 88ee607..139b1e4 100644
--- a/sts-agent/compose.yaml
+++ b/sts-agent/compose.yaml
@@ -1,6 +1,6 @@
 services:
   imds-emulator:
-    build: ../imds-emulator
+    build: .
     volumes:
       - /etc/sts-agent:/etc/sts-agent:ro
     environment:
diff --git a/imds-emulator/imds_emulator.py b/sts-agent/imds_emulator.py
similarity index 100%
rename from imds-emulator/imds_emulator.py
rename to sts-agent/imds_emulator.py
diff --git a/imds-emulator/imds_test.py b/sts-agent/imds_test.py
similarity index 100%
rename from imds-emulator/imds_test.py
rename to sts-agent/imds_test.py