diff --git a/command/maint_test.go b/command/maint_test.go
index 2793051306..0ac534e13a 100644
--- a/command/maint_test.go
+++ b/command/maint_test.go
@@ -45,12 +45,12 @@ func TestMaintCommandRun_NoArgs(t *testing.T) {
 	if err := a1.agent.AddService(service, nil, false, ""); err != nil {
 		t.Fatalf("err: %v", err)
 	}
-	if err := a1.agent.EnableServiceMaintenance("test", "broken 1"); err != nil {
+	if err := a1.agent.EnableServiceMaintenance("test", "broken 1", ""); err != nil {
 		t.Fatalf("err: %s", err)
 	}
 
 	// Enable node maintenance
-	a1.agent.EnableNodeMaintenance("broken 2")
+	a1.agent.EnableNodeMaintenance("broken 2", "")
 
 	// Run consul maint with no args (list mode)
 	ui := new(cli.MockUi)