diff --git a/tests/load/test_load.cpp b/tests/load/test_load.cpp
index 6fef0cf..72fd3a9 100644
--- a/tests/load/test_load.cpp
+++ b/tests/load/test_load.cpp
@@ -9,8 +9,13 @@
 #include "CppUTestExt/MockSupport.h"
 #include <cstdint>
 
+
+//This define allows us to dircetly include the device header without error.
+#define _AVR_IO_H_
+
 extern "C" 
 {
+#include <iotn404.h> //ATtiny404 header fille.
 #include "load.h"
 #include "MockADC.h"
 #include "MockADC.h"
@@ -33,3 +38,19 @@ TEST(test_load, LoadPass)
 {
     CHECK_TRUE(true);
 }
+
+TEST(test_load, PortAHandlerSuccess)
+{
+    mock().expectOneCall("ADC_Init")
+        .withUnsignedIntParameter("pin_num", 4);
+
+    mock().expectOneCall("ADC_Enable");
+    mock().expectOneCall("ADC_ReadValue_Impl")
+        .withUnsignedIntParameter("pin_num", 4);
+    mock().expectOneCall("ADC_Disable");
+    mock().expectOneCall("RegEdit_ClearBit")
+        .withPointerParameter("reg", (void *) &PORTA.OUT)
+        .withUnsignedIntParameter("bit_num", 7);
+    
+    Load_HandleLoadPortA(4, 7);
+}