diff --git a/plugin_test.go b/plugin_test.go
index 7a270eb..75e5904 100644
--- a/plugin_test.go
+++ b/plugin_test.go
@@ -8,9 +8,24 @@ import (
 	"encoding/json"
 	"testing"
 
+	"C"
+
 	"github.com/vanilla-os/vib/api"
 )
 
+type DnfModule struct {
+	Name    string     `json:"name"`
+	Type    string     `json:"type"`
+	Options DnfOptions `json:"options"`
+	Source  api.Source `json:"source"`
+}
+
+type DnfOptions struct {
+	Best       bool `json:"best"`
+	Refresh    bool `json:"refresh"`
+	SkipBroken bool `json:"skip_broken"`
+}
+
 type testCases struct {
 	module   DnfModule
 	expected string