diff --git a/client/allocrunner/taskrunner/task_runner_hooks.go b/client/allocrunner/taskrunner/task_runner_hooks.go
index 2e6fce183..6aa0da23a 100644
--- a/client/allocrunner/taskrunner/task_runner_hooks.go
+++ b/client/allocrunner/taskrunner/task_runner_hooks.go
@@ -218,9 +218,9 @@ func (tr *TaskRunner) prestart() error {
 
 		// Store the environment variables returned by the hook
 		if name == HookNameDevices {
-			tr.envBuilder.SetDeviceHookEnv(name, origHookState.Env)
+			tr.envBuilder.SetDeviceHookEnv(name, resp.Env)
 		} else {
-			tr.envBuilder.SetHookEnv(name, origHookState.Env)
+			tr.envBuilder.SetHookEnv(name, resp.Env)
 		}
 
 		// Store the resources