diff --git a/Testing/test.sh b/Testing/test.sh
index d4670fb..89b662d 100644
--- a/Testing/test.sh
+++ b/Testing/test.sh
@@ -39,8 +39,11 @@ download_formatter() {
     chmod +x task_formatter.sh
 
     # Source the formatter with error checking
-    if ! source ./task_formatter.sh; then
-        echo "Error: Failed to source task_formatter.sh"
+    source ./task_formatter.sh || { echo "Error: Failed to source task_formatter.sh"; exit 1; }
+    
+    # Check if print_header is available after sourcing
+    if ! declare -f print_header > /dev/null; then
+        echo "Error: print_header function not found after sourcing."
         exit 1
     fi
 }