1 files changed, 6 insertions, 5 deletions
diff --git a/cmd.go b/cmd.go
index 5e76708..4d56ad8 100644
--- a/cmd.go
+++ b/cmd.go
@@ -1,14 +1,15 @@
 package libfastimport
 
-import (
-	"git.lukeshu.com/go/libfastimport/textproto"
-)
-
 type fiReader interface {
 	PeekLine() (string, error)
 	ReadLine() (string, error)
 }
 
+type fiWriter interface {
+	WriteData(string) error
+	WriteLine(a ...interface{}) error
+}
+
 type cmdClass int
 
 const (
@@ -20,6 +21,6 @@ const (
 
 type Cmd interface {
 	fiCmdRead(fiReader) (Cmd, error)
-	fiCmdWrite(*textproto.FIWriter) error
+	fiCmdWrite(fiWriter) error
 	fiCmdClass() cmdClass
 }