diff --git a/platform/proxmox/tofu/main.tf b/platform/proxmox/tofu/main.tf
index 6734c7d..7d0eedf 100644
--- a/platform/proxmox/tofu/main.tf
+++ b/platform/proxmox/tofu/main.tf
@@ -19,6 +19,7 @@ resource "proxmox_virtual_environment_vm" "gpu" {
   node_name = each.value.node
   vm_id     = each.value.vmid
   tags      = ["knaster", "gpu", each.value.driver_generation]
+  pool_id   = try(each.value.pool, "gpu")
 
   machine       = "q35"
   bios          = "ovmf"
@@ -113,6 +114,7 @@ resource "proxmox_virtual_environment_container" "infra" {
   node_name     = each.value.node
   vm_id         = each.value.vmid
   tags          = ["knaster", "infra"]
+  pool_id       = try(each.value.pool, "infra")
   unprivileged  = true
   start_on_boot = true