diff --git a/deployment/data-model.nix b/deployment/data-model.nix
index 2c44544d..5a7ddd87 100644
--- a/deployment/data-model.nix
+++ b/deployment/data-model.nix
@@ -422,9 +422,8 @@ let
           };
           inherit nixos-configuration;
           ssh = host-ssh;
-          # TODO: add proxmox info
-          module = mkOption {
-            description = "The module to call to obtain the NixOS configuration from.";
+          caller = mkOption {
+            description = "The calling module to obtain the NixOS configuration from.";
             type = types.str;
           };
           args = mkOption {
@@ -506,7 +505,7 @@ let
                 inherit (tf-host.config)
                   system
                   ssh
-                  module
+                  caller
                   args
                   deployment-name
                   httpBackend
@@ -533,7 +532,7 @@ let
                 nixos_conf = writeConfig {
                   inherit
                     system
-                    module
+                    caller
                     args
                     deployment-name
                     root-path