diff --git a/client/allocrunner/taskrunner/task_runner_test.go b/client/allocrunner/taskrunner/task_runner_test.go
index c03d547e04f0ef2f7a6d3e8f8facfa0494f9edb8..8a6ab8cf50cb0076b6791550e2a0c5b86e21a867 100644
--- a/client/allocrunner/taskrunner/task_runner_test.go
+++ b/client/allocrunner/taskrunner/task_runner_test.go
@@ -2037,10 +2037,10 @@ func TestTaskRunner_BaseLabels(t *testing.T) {
 	}
 
 	config, cleanup := testTaskRunnerConfig(t, alloc, task.Name)
+	defer cleanup()
 
 	tr, err := NewTaskRunner(config)
 	require.NoError(err)
-	defer cleanup()
 
 	labels := map[string]string{}
 	for _, e := range tr.baseLabels {