diff --git a/plugin_test.go b/plugin_test.go
index eb14658..d14e770 100644
--- a/plugin_test.go
+++ b/plugin_test.go
@@ -94,7 +94,7 @@ func TestPlugin(t *testing.T) {
 			}
 
 			for _, tt := range tests {
-				g.Assert(tfApply(tt.args.config)).Equal(tt.want)
+				g.Assert(tfApply(tt.args.config).Args).Equal(tt.want.Args)
 			}
 		})
 	})