diff --git a/cmd/api/main.go b/cmd/api/main.go
index 6a7e0a6..01c0885 100644
--- a/cmd/api/main.go
+++ b/cmd/api/main.go
@@ -119,6 +119,7 @@ func main() {
 	router.Method("GET", endpoint.GetScheduledMessageEndpoint, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(scheduledMessageHandler.GetScheduledMessage)))
 	router.Method("GET", endpoint.FetchNextScheduledMessageEndpoint, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(scheduledMessageHandler.FetchNextMessage)))
 	router.Method("POST", endpoint.RecordEventResponse, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(eventHandler.RecordResponse)))
+	router.Method("GET", endpoint.FetchMessageEventResponse, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(eventHandler.Fetch)))
 	router.Method("POST", endpoint.SendInstantMessageEndpoint, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(instantMessageHandler.Send)))
 
 	router.Method("GET", "/swagger/*", httpSwagger.Handler(
diff --git a/cmd/api/main_test.go b/cmd/api/main_test.go
index d7f734f..953fff1 100644
--- a/cmd/api/main_test.go
+++ b/cmd/api/main_test.go
@@ -3,7 +3,6 @@ package main
 import (
 	"context"
 	"flag"
-	"fmt"
 	"log"
 	"net/http"
 	"os"
@@ -30,8 +29,8 @@ func TestMain(m *testing.M) {
 
 	database, err := db.NewDatabase(cfg.GetDBConnString())
 	if err != nil {
-		fmt.Println(err.Error())
-		panic("Failed to initialize database")
+		log.Println(err.Error())
+		log.Fatal("Failed to initialize database")
 	}
 	defer database.Close()
 
@@ -39,18 +38,18 @@ func TestMain(m *testing.M) {
 
 	if cfg.ResetDB {
 		if err = database.ResetDatabase(ctx); err != nil {
-			fmt.Println(err.Error())
-			panic("Failed to initialize database")
+			log.Println(err.Error())
+			log.Fatalln("Failed to initialize database")
 		}
 	} else {
 		if exists, err := db.TableExists(ctx, database.Pool, "contacts "); err == nil && !exists {
-			fmt.Println("Resetting database")
+			log.Println("Resetting database")
 			err = database.ResetDatabase(ctx)
 			if err != nil {
-				fmt.Printf("Error:%v", err)
+				log.Println("Error:", err)
 			}
 		} else {
-			fmt.Printf("Error:%v", err)
+			log.Println("Error:", err)
 		}
 	}
 
@@ -68,6 +67,7 @@ func TestMain(m *testing.M) {
 	scheduledMessageEventHandler := handler.NewScheduledMessageEventHandler(apiApp, schMsgEventStore, schStore)
 	scheduledMessageStatusHandler := handler.NewScheduledMessageStatusHandler(apiApp, schMsgEventStore, schStore)
 	eventHandler := handler.NewEventResponseHandler(apiApp, merStore)
+	instantMessageHandler := handler.NewSendInstantMessageHandler(apiApp, merStore, contactStore, messageStore)
 
 	testRouter = chi.NewRouter()
 	testRouter.Handle(endpoint.ADD_CONTACT_ENDPOINT, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(contactHandler.AddContact)))
@@ -81,6 +81,8 @@ func TestMain(m *testing.M) {
 	testRouter.Method("GET", endpoint.GetScheduledMessageEndpoint, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(scheduledMessageHandler.GetScheduledMessage)))
 	testRouter.Method("GET", endpoint.FetchNextScheduledMessageEndpoint, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(scheduledMessageHandler.FetchNextMessage)))
 	testRouter.Method("POST", endpoint.RecordEventResponse, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(eventHandler.RecordResponse)))
+	testRouter.Method("GET", endpoint.FetchMessageEventResponse, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(eventHandler.Fetch)))
+	testRouter.Method("POST", endpoint.SendInstantMessageEndpoint, mdlware.AuthMiddleware(jwtService)(http.HandlerFunc(instantMessageHandler.Send)))
 
 	code := m.Run()
 	os.Exit(code)