diff --git a/example_function_test.go b/example_function_test.go
index f61460e..dda890d 100644
--- a/example_function_test.go
+++ b/example_function_test.go
@@ -1,26 +1,31 @@
 package logrus_test
 
 import (
-	"fmt"
-	log "github.com/sirupsen/logrus"
 	"testing"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/stretchr/testify/assert"
 )
 
 func TestLogger_LogFn(t *testing.T) {
 	log.SetFormatter(&log.JSONFormatter{})
 	log.SetLevel(log.WarnLevel)
 
+	notCalled := 0
 	log.InfoFn(func() []interface{} {
-		fmt.Println("This is never run")
-		return []interface{} {
+		notCalled++
+		return []interface{}{
 			"Hello",
 		}
 	})
+	assert.Equal(t, 0, notCalled)
 
+	called := 0
 	log.ErrorFn(func() []interface{} {
-		fmt.Println("This runs")
-		return []interface{} {
+		called++
+		return []interface{}{
 			"Oopsi",
 		}
 	})
+	assert.Equal(t, 1, called)
 }