diff --git a/command/node_drain_test.go b/command/node_drain_test.go
index 6d7cc89d9..f75d0e503 100644
--- a/command/node_drain_test.go
+++ b/command/node_drain_test.go
@@ -59,8 +59,7 @@ func TestNodeDrainCommand_Detach(t *testing.T) {
 						Name:   "mock_task",
 						Driver: "mock_driver",
 						Config: map[string]interface{}{
-							"run_for":    "10m",
-							"exit_after": "10m",
+							"run_for": "10m",
 						},
 					},
 				},