diff --git a/src/device/input/wrap/refresh_test.go b/src/device/input/wrap/refresh_test.go
index e0ee5b8..165b8b7 100644
--- a/src/device/input/wrap/refresh_test.go
+++ b/src/device/input/wrap/refresh_test.go
@@ -2,7 +2,8 @@ package wrap
 
 import (
 	"context"
-	"mayhem-party/src/device/input/parse"
+	"mayhem-party/src/device/input/button"
+	"mayhem-party/src/device/input/raw"
 	"os"
 	"syscall"
 	"testing"
@@ -50,7 +51,7 @@ func TestRefreshDoesntCloseSources(t *testing.T) {
 	newParsers := 0
 	newParser := func() Wrap {
 		newParsers += 1
-		return parse.NewPlaintext(src)
+		return src
 	}
 	ctx, can := context.WithCancel(context.Background())
 	defer can()
@@ -72,7 +73,7 @@ func TestRefreshDoesntCloseSources(t *testing.T) {
 		ChSigUsr1 <- syscall.SIGINT
 	}
 	time.Sleep(time.Millisecond * 250)
-	if want := (telemetrySrc{reads: 5}); *src != want {
+	if want := (telemetrySrc{reads: 5, closeWraps: 5}); *src != want {
 		t.Errorf("want %+v, got %+v", want, *src)
 	} else if newParsers != 6 {
 		t.Error(newParsers)
@@ -80,15 +81,21 @@ func TestRefreshDoesntCloseSources(t *testing.T) {
 }
 
 type telemetrySrc struct {
-	closes int
-	reads  int
+	closeWraps int
+	closes     int
+	reads      int
+}
+
+func (src *telemetrySrc) CloseWrap() raw.Raw {
+	src.closeWraps += 1
+	return nil
 }
 
 func (src *telemetrySrc) Close() {
 	src.closes += 1
 }
 
-func (src *telemetrySrc) Read() []byte {
+func (src *telemetrySrc) Read() []button.Button {
 	src.reads += 1
-	return []byte("foo")
+	return nil
 }