diff --git a/tests/functional_tests/test_cases/serve/build_dag/test_call.py b/tests/functional_tests/test_cases/serve/build_dag/test_call.py
index fbbf8f885..9b93311eb 100644
--- a/tests/functional_tests/test_cases/serve/build_dag/test_call.py
+++ b/tests/functional_tests/test_cases/serve/build_dag/test_call.py
@@ -10,7 +10,7 @@ def test_generate_endpoint(self):
         response = requests.post(url, headers=headers, json=test_data)
 
         self.assertEqual(response.status_code, 200, 
-                        f"Expected status code 200, got {response.status_code}. Response: {response}")
+                        f"Expected status code 200, got {response.status_code}. Response: {response.json()}")
 
         self.assertGreater(len(response.text), 0,
                           "Generated text should not be empty")
diff --git a/tests/scripts/functional_tests/test_model.sh b/tests/scripts/functional_tests/test_model.sh
index c5714bb71..e63aa32dc 100755
--- a/tests/scripts/functional_tests/test_model.sh
+++ b/tests/scripts/functional_tests/test_model.sh
@@ -12,7 +12,6 @@ run_command() {
   return 0
 }
 
-
 source tests/scripts/_gpu_check.sh
 
 # Path to the YAML configuration file
@@ -119,6 +118,7 @@ test_model() {
           exit 1
         fi
       fi
+
       # Ensure that pytest check is completed before deleting the folder
       sleep 10s
     done