fix: test

Signed-off-by: Carlos A Becker <[email protected]>
charmbracelet · Dec 19, 2022 · b7c251c · b7c251c
1 parent 03c1e6b
commit b7c251c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main_test.go b/main_test.go
@@ -23,7 +23,7 @@ func TestMiddleware(t *testing.T) {
 	}
 
 	srv := &ssh.Server{
-		Handler: promwish.Middleware(listener.Addr().String(), "test", promwish.DefaultCommandFn)(func(s ssh.Session) {
+		Handler: promwish.Middleware(listener.Addr().String(), "test")(func(s ssh.Session) {
 			_, _ = s.Write([]byte("test"))
 			time.Sleep(500 * time.Millisecond)
 		}),