diff --git a/support/log/entry.go b/support/log/entry.go
index 9b3b596025..358d7d6163 100644
--- a/support/log/entry.go
+++ b/support/log/entry.go
@@ -40,6 +40,10 @@ func (e *Entry) SetLevel(level logrus.Level) {
 	e.entry.Logger.SetLevel(level)
 }
 
+func (e *Entry) SetExitFun(exitFun func(int)) {
+	e.entry.Logger.ExitFunc = exitFun
+}
+
 func (e *Entry) UseJSONFormatter() {
 	formatter := new(logrus.JSONFormatter)
 	formatter.TimestampFormat = timeStampFormat