diff --git a/socket_client.c b/socket_client.c
index ab0415d9..c6bb405c 100644
--- a/socket_client.c
+++ b/socket_client.c
@@ -60,7 +60,7 @@ int main (int argc, char **argv) {
 	while (strstr(buffer, "--EOM--") == NULL)
 	{
 
-		printf("Message received: \"%s\"\n", buffer);
+		printf("%s", buffer);
 
 		// Receive message
 		size = recv(socketfd, buffer, BUF-1, 0);
@@ -70,7 +70,7 @@ int main (int argc, char **argv) {
 			buffer[size] = '\0';
 	}
 
-	printf("Message received: \"%s\"\n", buffer);
+	printf("%s", buffer);
 
 	close (socketfd);
 	return EXIT_SUCCESS;
diff --git a/test/test_suite.sh b/test/test_suite.sh
index a4907c1f..4ca4ca29 100644
--- a/test/test_suite.sh
+++ b/test/test_suite.sh
@@ -197,9 +197,11 @@ load 'libs/bats-support/load'
 }
 
 @test "Unix socket returning data" {
-  run bash -c './pihole-FTL help'
+  run bash -c './socket-test travis'
   echo "output: ${lines[@]}"
-  [[ ${lines[0]} == "pihole-FTL - The Pi-hole FTL engine" ]]
+  [[ ${lines[0]} == "Socket created" ]]
+  [[ ${lines[1]} == "Connection established" ]]
+  [[ ${lines[2]} ~= "domains_being_blocked -1" ]]
 }
 
 @test "Final part of the tests: Killing pihole-FTL process" {