diff --git a/tools/nats_caller.go b/tools/nats_caller.go
index adc6af3..b5ba946 100644
--- a/tools/nats_caller.go
+++ b/tools/nats_caller.go
@@ -92,7 +92,7 @@ func (s *natsCaller) ListenNats(execs map[NATSMethod]func(map[string]string)) {
 }
 
 // SetNATSPub sets a message to the NATS server
-func (o *natsCaller) SetNATSPub(method NATSMethod, data interface{}) string {
+func (o *natsCaller) SetNATSPub(method NATSMethod, data NATSResponse) string {
 	if config.GetConfig().NATSUrl == "" {
 		return " -> NATS_SERVER is not set"
 	}